pytorch求幂次方函数

PyTorch中可以使用torch.pow函数来求幂次方。它的用法如下：

torch.pow(input, exponent, out=None) -> Tensor

其中，input是输入的tensor，exponent是幂次方的指数，out是可选的输出tensor。

举个例子，如果要计算2的3次方，可以这样写：

import torch

x = torch.tensor(2.0)
y = torch.pow(x, 3)
print(y)

输出结果为：

tensor(8.)

如果要计算一个矩阵的幂次方，可以这样写：

import torch

x = torch.tensor([[1, 2], [3, 4]], dtype=torch.float32)
y = torch.pow(x, 2)
print(y)

输出结果为：

tensor([[ 1.,  4.],
        [ 9., 16.]])