Pytorch中randn和rand函数的用法

Pytorch中randn和rand函数的用法

  • randn

torch.randn(*sizes, out=None) → Tensor

返回一个包含了从标准正态分布中抽取的一组随机数的张量

size:张量的形状,

out:结果张量。(目前还没有看到使用这个参数的例子)

  • rand也差不多其实:

torch.rand(*sizes, out=None) → Tensor

但是它是[0,1)之间的均匀分布

  • 其他一些分布

离散正态分布
torch.normal(means, std, out=None) → → Tensor

线性间距向量
torch.linspace(start, end, steps=100, out=None) → Tensor


reference:

https://blog.csdn.net/Hningning/article/details/102575272

原文地址:https://www.cnblogs.com/jiading/p/11944458.html