Element wise sum of batched tensors

[ptrblck]

Your approach might work, but if you want to set a dim argument you could use torch.sum(t_list, dim=0).