torch随机生成整数
时间: 2023-06-05 09:47:27 浏览: 620
Pytorch生成随机数Tensor的方法汇总
5星 · 资源好评率100%
torch是一种流行的Python机器学习库,其中包含了许多用于生成随机数的函数。如果需要生成随机整数,可以使用torch.randint()函数,该函数可以用来随机生成一个整数的张量。
torch.randint()函数需要指定整数的范围和张量的形状,例如:
```
import torch
random_ints = torch.randint(0, 10, (3,3))
print(random_ints)
```
这将生成一个3x3的张量,其中所有的元素都是[0,10)之间的随机整数。如果需要生成更大的整数,则可以适当调整范围。如果需要生成负数,则可以在范围内添加一个负数值。
此外,torch还提供了许多其他的随机数生成函数,例如torch.rand()(生成[0,1)之间的随机数)、torch.randn()(生成标准正态分布中的随机数)等等。根据不同的需求,可以选择不同的函数来生成合适的随机数。
阅读全文