为什么使用表 {1},{{1}} 索引 torch.Tensor 会得到不同的结果?

假设 A 是一个 torch 张量,大小为 3x10x10,在使用表 {1},{{1}} 和 {{{1}}} 进行索引时为什么会得到不同的结果?

In [53]: A:size()
Out[53]:
  3
 10
 10
[torch.LongStorage of size 3]

In [54]: A[{1}]:size()
'Out[54]:
 10
 10
[torch.LongStorage of size 2]

In [55]: A[{{1}}]:size()
Out[55]:
  1
 10
 10
[torch.LongStorage of size 3]

In [56]: A[{{{1}}}]:size()
Out[56]:
  3
 10
 10
[torch.LongStorage of size 3]

点赞