火炬/ Lua元素明智地乘以2D和1D张量

Question

我正在尝试在2D批量张量（128x512）和1D张量（512）之间预先形成元素乘法。

目前，我正在这样做：

   nbatch = input:size(1)
   for i = 1 , nbatch , 1 do
      self.output[i]:cmul(self.noise)
   end

它有效并且我得到了预期的结果，但我认为这不是效率最高的原因。

可以更有效地完成吗？

如何将nD张量元素乘以（n-1）D张量进行扩展？

谢谢！

Answer 1

self.output:cmul(self.noise:view(1, self.output:size(2)):expandAs(self.output))