對火炬中的每一行應用函數

Question

我知道張量有一個apply方法，但是這隻適用於每個元素的函數。有沒有一種優雅的方式來進行按行操作？例如，我可以乘以不同的值每行？

說

A = 
    1 2 3 
    4 5 6 
    7 8 9 

和

B = 
    1 
    2 
    3 

，我想通過B的i個元素乘以A的i第i行中的每個元素得到

1 2 3 
8 10 12 
21 24 27 

我會怎麼做？

Answer 1

一種可能性是expandB如下：

1 1 1 
2 2 2 
3 3 3 
[torch.DoubleTensor of size 3x3] 

然後你就可以直接使用element-wise multiplication：

local A = torch.Tensor{{1,2,3},{4,5,6},{7,8,9}} 
local B = torch.Tensor{1,2,3} 
local C = A:cmul(B:view(3,1):expand(3,3)) 

Answer 2

請參閱此鏈接：Torch - Apply function over dimension

（感謝亞歷山大Lutsenko的提供它，我剛把它移到答案上。）