lua - Torch 教程：1_data.lua 中“trainData.data[{ {},i,{},{} }]:mean()”的含义

Question

在火炬教程中，我找到了这一行：

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

有没有人可以解释索引 { {},i,{},{} } 在做什么？我可以猜到，但想知道确切的机制。

提前致谢。

Answer 1

这实际上是张量缩小/切片的简洁语法，在文档中有详细说明。

在 中[{ ... }]，您可以为张量的每个维度：

• 传递一个数字n以仅沿此维度保留n-th 组件，
• 传递一个范围{start,end}以保持所有组件从start到end沿着这个维度，
• 通过{}以保持所有组件沿此维度。

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