Lua/torch 1D 和 2D 张量的乘法

我正在尝试在 Lua 中对两个矩阵进行乘法,它们的维度分别为 a=40,000x1 和 b=1x40,000。在 Lua 中,40,000x1 矩阵显示为 1D 张量,而 1x40,000 矩阵显示为 2D 张量。每当我尝试使用 a*b 简单地进行乘法运算时,会出现错误: multiplication between 1D and 2D tensors not yet supported。我不能迭代地遍历每个索引,因为这个函数在我的程序中经常使用,会大大增加执行时间。我该怎么乘 ab 呢?

点赞
用户1688185
用户1688185

使用 view

c = a:view(40000, 1) * b
2016-06-06 19:58:13