torch.randn
时间: 2023-08-20 18:14:48 浏览: 86
torch.randn是一个用来生成随机数字的函数,这些随机数字满足标准正态分布(0~1)。它可以接受一个整数或一个元组作为参数来指定生成的随机数字的维度。[2] 例如,torch.randn(3)会生成一个包含3个随机数字的一维张量,而torch.randn(3,4)会生成一个包含3行4列的二维张量。[2] 你可以使用print函数来打印生成的随机张量。例如,对于torch.randn(3),你可以使用print("a:",a)来打印生成的随机一维张量a。[2]
阅读全文