ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

torch.normal()

2021-04-08 11:05:14  阅读:11  来源: 互联网

标签:normal True torch requires grad size


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)

 



标签:normal,True,torch,requires,grad,size
来源: https://www.cnblogs.com/qclan/p/14631192.html

专注分享技术,共同学习,共同进步。侵权联系[admin#icode9.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有