torch.normal()


torch.normal(A, B ,size(C, D), requires_grad=True)

A表示均值,B表示标准差 ,C代表生成的数据行数,D表示列数,requires_grad=True表示对导数开始记录,可以忽略。

1 import torch
2 w = torch.normal(1, 0.02, size=(3, 1), requires_grad=True)
3 print(w)

得到的结果为:

tensor([[0.9850],
        [0.9749],
        [1.0409]], requires_grad=True)

 

 



					


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM