如何在 torch 中计算两个张量表的逐一对应加法

我在我的网络(深度神经网络)中有两层 L1 和 L2,它们输出大小为 m 和 n 的张量表 T1 和 T2(这意味着 T1 和 T2 分别具有大小为 m 和 n 的张量)。

现在我想将 T1 和 T2 的每一对张量相加,以便结果表具有 $m \times n$ 相同大小的张量。

如何在 torch 中实现?

点赞
用户1984680
用户1984680

如果我理解正确的话,只需要使用nn.CAddTable类。

require 'nn'

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

m = nn.CAddTable()
m:forward({T1,T2})
2017-04-12 20:44:02