1D和2D张量的Lua /火炬倍增

时间:2016-06-06 19:51:36

标签: matrix lua torch

我试图在lua中乘以两个矩阵,其尺寸为a = 40,000x1和b = 1x40,000。在Lua中,40,000x1矩阵显示为1D张量,1x40,000矩阵显示为2D张量。无论何时,我只是使用* b尝试将它们相乘,出现错误:multiplication between 1D and 2D tensors not yet supported。我不能迭代地遍历每个索引,因为这个函数在我的程序中经常使用,并且会大大增加执行时间。如何将ab相乘?

1 个答案:

答案 0 :(得分:2)

使用view

c = a:view(40000, 1) * b