torch.nn.Init.normal_()的用法

torch.nn.init.normal(tensor, mean=0, std=1)
从给定均值和标准差的正态分布N(mean, std)中生成值,填充输入的张量或变量

参数:
tensor – n维的torch.Tensor
mean – 正态分布的均值
std – 正态分布的标准差

posted @ 2021-03-15 20:05  吴莫愁258  阅读(13912)  评论(0编辑  收藏  举报