在火炬教程中,我发现了这一行:
mean[i] = trainData.data[{ {},i,{},{} }]:mean()
是否有人可以解释索引{{},i,{},{}}正在做什么? 我猜,但想知道确切的机制。
提前致谢。
答案 0 :(得分:3)
这实际上是张量缩小/切片的简明语法,详见文档中的here。
在[{ ... }]
内,您可以为张量的每个维度:
n
号码,只保留此n
个组件,{start,end}
以保持start
到end
的所有组件沿此维度,{}
以保留此维度的所有组件。在这个精确的情况下,它是从u * v * w * x
张量到u * 1 * w * x
张量的缩小,只保留第二维上的i
- 分量。