pytorch是否支持多张量CAddTable?
Does pytorch support multiple tensor CAddTable?
我在 official doc 中发现 CAddTable 应该按照
x = x1 + x2 # instead of CAddTable(x1, x2) in older version
PyTorch 会做其他事情,比如 autograd
但是如果我有多个张量呢?将上面的输入从两个张量更改为张量列表。 PyTorch 还能做类似的事情吗?
只是为了清楚地显示评论中的代码片段:
x = torch.stack((x1, x2, x3, x4), dim=0)
y = torch.sum(x, dim=0, keepdim=False) # same shape as x1, x2...
我在 official doc 中发现 CAddTable 应该按照
x = x1 + x2 # instead of CAddTable(x1, x2) in older version
PyTorch 会做其他事情,比如 autograd
但是如果我有多个张量呢?将上面的输入从两个张量更改为张量列表。 PyTorch 还能做类似的事情吗?
只是为了清楚地显示评论中的代码片段:
x = torch.stack((x1, x2, x3, x4), dim=0)
y = torch.sum(x, dim=0, keepdim=False) # same shape as x1, x2...