Pytorch向量各个元素的n次方求和

        代码

import torch

# set torch print options preference
torch.set_printoptions(linewidth=1024)

# input
a = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
print(a)

# calculate
p = (a ** 5).sum(-1).item()

# output
print(p)

        输出

tensor([1., 2., 3., 4., 5.])
4425.0