如何在火炬中计算两张张量的成对加法

Question

我的网络（深度神经网络）L1和L2中有两层输出张量表。假设输出是大小为m和n的表T1和T2（这意味着T2 & T2分别具有m和n张量）。

现在我想要添加T1和T2的每一对张量，以便生成的表具有相同大小的$ m \ times n $张量。

如何在火炬中做到这一点？

Answer 1

如果我理解正确，只需使用nn.CAddTable类。

require 'nn' 

T1 = torch.Tensor(20,10) 
T2 = torch.Tensor(20,10) 

m = nn.CAddTable() 
m:forward({T1,T2})