乘张量lua

Question

我是lua的新手，

为什么以下代码不起作用？

local data = torch.Tensor(100, 4)
--data tensor is read from file
local w = torch.Tensor(1, 4):zero()
local tmp = w * data[5]:t()

Answer 1

data[5]是一维张量，transpose仅适用于二维张量。因此，您可以按照以下indexing运算符获取额外维度：

 -- matrix-matrix operation: result is a 1x1 tensor
 local tmp = w * data[{{5}}]:t()

或者squeeze w的第一个单例维度：

 -- dot prod between 1D tensors: result is a number
 local tmp = w:squeeze() * data[5]