hey guys, Iā m new to pytorch, I just want to know is there any pytorch API that
can process the tensor with l2-normalization? In tensorflow, the corresponding API is tf.nn.l2_normalize.
1 Like
I think I just got the answer.
- import torch.nn.functional as f
- a = torch.randn(2,3)
- norm_a = f.normalize(a,dim=0,p=2)
where p=2 means the l2-normalization, and dim=0 means normalize tensor a with row.
4 Likes
Thanks for the code. It should be -1 instead of 0 I think.
norm_a = f.normalize(a,dim=-1,p=2)