torch.norm() と F.normalize() の違い

2つの違い

F.normalize() は torch.norm() に基づいて標準化されています。つまり、次のようになります。

F.normalize(x) = x / torch.norm(x)

コード例

import torch
import torch.nn.functional as F


x = torch.tensor([[2.3]]).float()
r1 = x.norm(2)
r2 = F.normalize(x)

## 结果
r1: tensor(3.6056)
r2: tensor([[0.5547, 0.8321]]) = tensor([[2., 3.]]) / tensor(3.6056)