pytorch张量求绝对值
时间: 2023-11-20 20:54:13 浏览: 274
Python中,求绝对值.md
可以使用torch.abs函数来求解PyTorch张量的绝对值。该函数的原型为torch.abs(input, *, out=None) → Tensor,其中input为输入的张量,out为输出的张量(可选参数)。下面是一个例子:
```
import torch
x = torch.tensor([-1, -2, 3])
y = torch.abs(x)
print(y)
```
输出结果为:tensor([1, 2, 3])
此外,还可以使用绝对值函数的另一种形式,即使用torch.abs_()函数直接对原始张量进行修改。例如:
```
import torch
x = torch.tensor([-1, -2, 3])
x.abs_()
print(x)
```
输出结果为:tensor([1, 2, 3])
阅读全文