火炬教程:" trainData.data [{{},i,{},{}}]的含义:mean()"在1_data.lua中

时间:2015-07-17 14:28:22

标签: lua torch

在火炬教程中,我发现了这一行:

mean[i] = trainData.data[{ {},i,{},{} }]:mean()

是否有人可以解释索引{{},i,{},{}}正在做什么? 我猜,但想知道确切的机制。

提前致谢。

1 个答案:

答案 0 :(得分:3)

这实际上是张量缩小/切片的简明语法,详见文档中的here

[{ ... }]内,您可以为张量的每个维度:

  • 传递一个n号码,只保留此n个组件,
  • 传递范围{start,end}以保持startend的所有组件沿此维度,
  • 传递{}以保留此维度的所有组件。

在这个精确的情况下,它是从u * v * w * x张量到u * 1 * w * x张量的缩小,只保留第二维上的i - 分量。