随机生成torch.tensor
时间: 2023-11-19 15:47:00 浏览: 62
Pytorch生成随机数Tensor的方法汇总
torch.rand(*sizes, out=None)是一个函数,用于生成一个随机的tensor,其中tensor的形状由sizes定义。这个函数会返回一个tensor,其数值是从[0,1)的均匀分布中抽取的一组随机数。另外,torch.randn(*sizes, out=None)是一个类似的函数,不同之处在于它从标准正态分布中抽取随机数。你也可以使用torch.normal()来生成从指定均值和标准差的正态分布中抽取的随机数。还有一个函数是torch.linspace(),它可以生成一个区间内均匀间隔的数值序列。
下面是一个示例代码,用于生成一个5x3的随机tensor:
x = torch.rand(5,3)
print(x)
阅读全文