2D 및 1D 텐서의 Torch/Lua 엘리먼트 현명한 곱셈

Question

2D 배치 텐서 (128x512)와 1D 텐서 (512) 사이의 요소 현명한 곱셈을 수행하려고합니다. 그것은 작동하고 내가 기대했던 결과를 얻을 수 있지만, 내가 왜 그것을 할 가장 효율적인 아니라고 생각

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

: 왜

현재,이 그것을하고 있어요.

더 효율적으로 수행 할 수 있습니까?

(n-1) D tensors로 nD tensors 요소를 곱하면 어떻게 확장 할 수 있습니까?

감사합니다.

Answer 1

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