我试图在lua中乘两个矩阵,其维数是a=40,000x1和b=1x40,000。在Lua中,40000x1矩阵显示为一维张量,1x40000矩阵显示为2D张量。每当我试图简单地使用a*b乘以它们时,就会出现一个错误:multiplication between 1D and 2D tensors not yet supported。我不能迭代地遍历每个索引,因为这个函数在我的程序中经常使用,并且会大大增加执行时间。如何将a和b相乘
操作:我有维度[n x m x c]的pytorch张量A和维度[1 x 1 x c]的B。我想将每个1 x 1 x c向量的内积与A的B相乘,从而生成[n x m]维的张量C。在我的网络的前向函数中,在特定的步骤我接收维度[N, channels, Height, Width]的张量,其中N是图像的数量,通道是特征映射中的通道的数量,高度和宽度是当前特征映射的。有人能解释一下pytorch中实现这一步的最佳方式和功能吗? 我刚接触pytorch,找不到合适的方法。Te