如何计算火炬中两个张量表的成对相加

Question

我的网络(深度神经网络(L1和L2中有两层，它们输出张量表。假设输出是大小为 m 和 n 的表 T1 和 T2(这意味着 T1 和 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})