提交 a273e361 authored 作者: Frederic Bastien's avatar Frederic Bastien

tell subensor view are supported on GPU

上级 b7c20360
...@@ -394,14 +394,14 @@ e = a + d ...@@ -394,14 +394,14 @@ e = a + d
stringstyle=\color{violet}, stringstyle=\color{violet},
} }
\begin{lstlisting} \begin{lstlisting}
# return views # return views, supported on GPU
a_tensor[int] a_tensor[int]
a_tensor[int, int] a_tensor[int, int]
a_tensor[start:stop:step, start:stop:step] a_tensor[start:stop:step, start:stop:step]
a_tensor[::-1] # reverse the first dimension a_tensor[::-1] # reverse the first dimension
# Advanced indexing, return copy # Advanced indexing, return copy
a_tensor[an_index_vector] # Supported on GPU. a_tensor[an_index_vector] # Supported on GPU
a_tensor[an_index_vector, an_index_vector] a_tensor[an_index_vector, an_index_vector]
a_tensor[int, an_index_vector] a_tensor[int, an_index_vector]
a_tensor[an_index_tensor, ...] a_tensor[an_index_tensor, ...]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论