torch.nn.init.constant_(tensor, val)如何使用

這篇文章主要講解了“torch.nn.init.constant_(tensor, val)如何使用”，文中的講解內容簡單清晰，易于學習與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學習“torch.nn.init.constant_(tensor, val)如何使用”吧！

功能:使用val的值來填充輸入的Tensor

代碼實驗展示:

Microsoft Windows [版本 10.0.18363.1256](c) 2019 Microsoft Corporation。保留所有權利。

C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_2_0) C:\Users\chenxuqi>python
Python 3.7.7 (default, May  6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> import torch.nn as nn>>> w = torch.empty(3, 2)>>> w
tensor([[1.0381e+23, 4.4141e-43],[1.0381e+23, 4.4141e-43],[1.0381e+23, 4.4141e-43]])>>> nn.init.constant_(w, 20200910.0)tensor([[20200910., 20200910.],[20200910., 20200910.],[20200910., 20200910.]])>>>>>>>>> weight = nn.Parameter(torch.Tensor(3))>>> weight
Parameter containing:tensor([0.0000e+00, 0.0000e+00, 2.9066e+38], requires_grad=True)>>>>>> nn.init.constant_(weight, 20200910.0)Parameter containing:tensor([20200910., 20200910., 20200910.], requires_grad=True)>>> weight
Parameter containing:tensor([20200910., 20200910., 20200910.], requires_grad=True)>>>>>>>>>

感謝各位的閱讀，以上就是“torch.nn.init.constant_(tensor, val)如何使用”的內容了，經過本文的學習后，相信大家對torch.nn.init.constant_(tensor, val)如何使用這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！