以函数 nn.init.xavier_uniform_()和 nn.init.xavier_uniform()为例
带有下划线的nn.init.xavier_uniform_()在一个tensor上操作了之后,直接修改这个tensor,没有则不会修改。
pytorch中函数名有下划线和没有下划线
最新推荐文章于 2026-01-05 17:06:32 发布
这篇博客探讨了PyTorch中nn.init.xavier_uniform_()和nn.init.xavier_uniform()两个函数的区别。nn.init.xavier_uniform_()是一个就地操作,直接修改输入Tensor的值,而nn.init.xavier_uniform()则不修改原始Tensor。理解这些初始化方法对于深度学习模型的权重初始化至关重要,影响模型的训练效果。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.6
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
4356

被折叠的 条评论
为什么被折叠?



