2020年12月15日
pytorch normal_(), fill_()
摘要： 比如有个张量a，那么a.normal_()就表示用标准正态分布填充a，是in_place操作，如下图所示： 比如有个张量b，那么b.fill_(0)就表示用0填充b，是in_place操作，如下图所示： 这两个函数常常用在神经网络模型参数的初始化中，例如 import torch.nn as nn 阅读全文
posted @ 2020-12-15 17:37 Picassooo 阅读(8856) 评论(0) 推荐(0)