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

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

功能:使用val的值來(lái)填充輸入的Tensor

代碼實(shí)驗(yàn)展示:

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

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)如何使用”的內(nèi)容了，經(jīng)過(guò)本文的學(xué)習(xí)后，相信大家對(duì)torch.nn.init.constant_(tensor, val)如何使用這一問(wèn)題有了更深刻的體會(huì)，具體使用情況還需要大家實(shí)踐驗(yàn)證。這里是億速云，小編將為大家推送更多相關(guān)知識(shí)點(diǎn)的文章，歡迎關(guān)注！