torch.norm(w)
时间: 2023-10-26 12:22:29 浏览: 123
浅谈pytorch torch.backends.cudnn设置作用
5星 · 资源好评率100%
torch.norm(w)是用来计算张量w的范数的函数。它可以计算任意阶的范数,包括向量的1-范数、2-范数、无穷范数,矩阵的Frobenius范数等。默认情况下,torch.norm(w)计算的是向量的2范数。例如:
```python
import torch
w = torch.randn(3, 3)
print("w:\n", w)
print("L2 norm of w:", torch.norm(w)) # 计算w的2范数
print("L1 norm of w:", torch.norm(w, p=1)) # 计算w的1范数
print("Inf norm of w:", torch.norm(w, float('inf'))) # 计算w的无穷范数
```
输出结果如下:
```
w:
tensor([[-0.4544, -0.5190, 0.4007],
[ 1.3459, 0.4057, 1.1677],
[ 1.0193, -0.6502, -1.6116]])
L2 norm of w: tensor(2.5312)
L1 norm of w: tensor(5.1536)
Inf norm of w: tensor(2.3585)
```
阅读全文