pytorch的归一化函数有哪些
时间: 2024-02-05 12:01:21 浏览: 83
pytorch方法测试详解——归一化(BatchNorm2d)
5星 · 资源好评率100%
PyTorch的归一化函数主要有两个:torch.nn.functional.normalize()和torch.nn.BatchNorm1d()。其中,torch.nn.functional.normalize()用于将向量标准化,torch.nn.BatchNorm1d()用于在训练过程中对1D输入进行批量归一化。
阅读全文