【PyTorch】在Tensor上的索引和切片
一般索引
从前往后提供的索引,会依次在每个维度上做索引。
import torch
a = torch.rand(4, 3, 28, 28)
print(a[0].shape)
print(a[0, 0].shape)
print(a[0, 0, 2, 4]) # 具体到某个元素
运行结果:
torch.Size([3, 28, 28])
torch.Size([28, 28])
...
转载
2020-04-08 11:52:44 ·
1064 阅读 ·
1 评论