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)

 



上一篇:梯度计算


下一篇:0407手动实现线性回归