torch.linalg.norm
时间: 2023-10-18 15:55:57 浏览: 236
PyTorch里面的torch.nn.Parameter()详解
3星 · 编辑精心推荐
torch.linalg.norm 是 PyTorch 中的一个函数,用于计算张量的范数。它可以计算向量范数、矩阵范数以及张量范数。
用法示例:
```python
import torch
# 计算向量范数
v = torch.tensor([1, 2, 3])
norm_v = torch.linalg.norm(v)
print(norm_v) # 输出: tensor(3.7417)
# 计算矩阵范数
m = torch.tensor([[1, 2], [3, 4]])
norm_m = torch.linalg.norm(m)
print(norm_m) # 输出: tensor(5.4772)
# 计算张量范数
t = torch.tensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
norm_t = torch.linalg.norm(t)
print(norm_t) # 输出: tensor(14.2829)
```
在计算范数时,可以指定不同的参数来计算不同的范数,比如计算二范数、一范数、无穷范数等。具体可以参考 PyTorch 文档中 torch.linalg.norm 的说明。
阅读全文