继续(机器)翻译:)
torch.normal — PyTorch 1.10.1 documentation
torch.normal(mean, std, *, generator=None, out=None) → Tensor
返回一个从独立的正态分布中抽取的随机数的张量,正态分布的平均值为mean、标准差为std。
mean是一个张量,包含每个输出元素的正态分布的平均值。
std是一个张量,包含每个输出元素的正态分布的标准差。
平均值mean和标准差std的形状不需要匹配,但这两个张量中的元素总数需要相同。当他俩形状不匹配时,使用平均值的形状作为输出张量的形状。
When std is a CUDA tensor, this function synchronizes its device with the CPU.