这在Torch Tensor Documentation中的索引中有所涉及
使用空表{} 进行索引是该维度中所有索引的简写。下面是一个使用{} 将整行从一个矩阵复制到另一个矩阵的演示:
> a = torch.Tensor(3, 3):fill(0)
0 0 0
0 0 0
0 0 0
> b = torch.Tensor(3, 3)
> for i=1,3 do for j=1,3 do b[i][j] = (i - 1) * 3 + j end end
> b
1 2 3
4 5 6
7 8 9
> a[{1, {}}] = b[{3, {}}]
> a
7 8 9
0 0 0
0 0 0
这个赋值相当于:a[1] = b[3]。
你的例子很相似:
im4[{1,{},{}}] = im3[{3,{},{}}]
im4[{3,{},{}}] = im3[{1,{},{}}]
更清楚的表述为:
im4[1] = im3[3]
im4[3] = im3[1]
第一行将im3的第三行(二维子矩阵)中的值分配给im4的第一行,第二行将im3的第一行分配给@的第三行987654331@.
请注意,这不是交换,因为永远不会写入 im3,永远不会读取 im4。