以如下 tensor a
为例,展示常用的 indxing
, slicing
及其他高阶操作
>>> a = torch.rand(4,3,28,28)
>>> a.shape
torch.Size([4, 3, 28, 28])
-
Indexing: 使用索引获取目标对象,
[x,x,x,....]
>>> a[0].shape torch.Size([3, 28
以如下 tensor a
为例,展示常用的 indxing
, slicing
及其他高阶操作
>>> a = torch.rand(4,3,28,28)
>>> a.shape
torch.Size([4, 3, 28, 28])
Indexing: 使用索引获取目标对象,[x,x,x,....]
>>> a[0].shape
torch.Size([3, 28