pytorch 求绝对值
时间: 2023-10-18 21:26:16 浏览: 124
Python中,求绝对值.md
在 PyTorch 中,可以使用 `torch.abs()` 函数来求取张量的绝对值。下面是一个例子:
```python
import torch
x = torch.tensor([-1, 2, -3])
abs_x = torch.abs(x)
print(abs_x)
```
输出结果为:
```
tensor([1, 2, 3])
```
通过调用 `torch.abs()` 函数,我们可以得到输入张量 `x` 中每个元素的绝对值。
阅读全文