CODEWITHSUNDEEP
pytorch
Litchy
Litchy

Reputation: 673

Does pytorch support multiple tensor CAddTable?

I found in official doc that CAddTable should be done as

x = x1 + x2 # instead of CAddTable(x1, x2) in older version

and PyTorch would do the rest of things like autograd

But how about if I have multiple tensors, aka. changing the input above from two tensors to a list of tensors. Could PyTorch still do the similar things?

Upvotes: 0

Views: 41

Answers (1)

cao-nv
cao-nv

Reputation: 384

Just for a clean display of the code snip in the comment:

x = torch.stack((x1, x2, x3, x4), dim=0)
y = torch.sum(x, dim=0, keepdim=False) # same shape as x1, x2...

Upvotes: 1

Related Questions