pytorch- normal_用法

import torch
x = torch.zeros((2, 3))
x.normal_(0, 1)

tensor([[-0.8139,  0.6736,  0.3647],
        [ 0.3843, -0.2469, -0.9998]])

normal(mean, variance)
weight 初始化的正态分布。

你可能感兴趣的:(Python,deep,learning)