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)