python幂函数pytorch

在Pytorch中，可以使用torch.pow()函数来计算幂函数。该函数的作用是将输入张量的每个元素都进行幂运算，并返回一个新的张量。

使用方法如下：
torch.pow(input, exponent, out=None)

其中，input是待求幂的张量，exponent是幂指数。可以通过设置out参数来指定输出的张量。

以下是一个简单的示例，说明如何使用torch.pow()函数：
import torch

x = torch.tensor([2, 3, 4])
exponent = 2
result = torch.pow(x, exponent)
print(result)

输出结果为：
tensor([ 4, 9, 16])