Lua/torch 一维和二维张量的乘法
Lua/torch multiplication of 1D and 2D tensors
我正在尝试将 lua 中的两个矩阵相乘,其尺寸为 a=40,000x1 和 b=1x40,000。在 Lua 中,40,000x1 矩阵显示为一维张量,1x40,000 矩阵显示为二维张量。每当我尝试简单地使用 a*b 将它们相乘时,就会出现一个错误:multiplication between 1D and 2D tensors not yet supported
。我不能迭代地遍历每个索引,因为这个函数在我的程序中经常使用,并且会大大增加执行时间。我怎样才能乘以 a
和 b
?
使用view:
c = a:view(40000, 1) * b
我正在尝试将 lua 中的两个矩阵相乘,其尺寸为 a=40,000x1 和 b=1x40,000。在 Lua 中,40,000x1 矩阵显示为一维张量,1x40,000 矩阵显示为二维张量。每当我尝试简单地使用 a*b 将它们相乘时,就会出现一个错误:multiplication between 1D and 2D tensors not yet supported
。我不能迭代地遍历每个索引,因为这个函数在我的程序中经常使用,并且会大大增加执行时间。我怎样才能乘以 a
和 b
?
使用view:
c = a:view(40000, 1) * b