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)

 




posted @ 2021-04-08 10:54  qclan  阅读(5278)  评论(0)    收藏  举报