individual_sum = list(map(torch.sum, group)) # You can use loop/list comprehension etc. 9. # Output. 10. # [tensor(0.), tensor(4.) ...
確定! 回上一頁