torch.pow
时间: 2023-07-03 13:06:39 浏览: 229
pyTorch.randn()、rand()、cat()、pow()、scatter_()、.squeeze() 、.unsqueeze()
5星 · 资源好评率100%
torch.pow() 是PyTorch中的一个函数,它用于计算输入张量的指数幂。它的语法如下:
```python
torch.pow(input, exponent, out=None)
```
其中,`input`是输入张量,`exponent`是指数,`out`是输出张量(可选参数)。返回值是一个新的张量,其元素为`input`的对应元素的指数幂。
例如,我们可以这样使用 torch.pow():
```python
import torch
x = torch.tensor([1, 2, 3])
y = torch.pow(x, 2)
print(y)
```
这将输出:
```
tensor([1, 4, 9])
```
在这个例子中,我们将输入张量x的每个元素平方,并将结果存储在输出张量y中。
阅读全文