- 函数作用
对输入的数据（tensor）进行指定维度的L2_norm运算。
- 代码示例
import torch.nn.functional as F
>>> a=torch.tensor([1,2,3,4],dtype=torch.float32)
>>> a
tensor([1., 2., 3., 4.])
>>> F.normalize(a,p=2,dim=0)
#对指定维度进行运算
tensor([0.1826, 0.3651, 0.5477, 0.7303])
最后
以上就是幸福冬日最近收集整理的关于torch.nn.functional.normalize()函数解读的全部内容，更多相关torch内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复