torch 取随机自然数
时间: 2023-09-03 10:05:28 浏览: 144
作为一个 AI,我不能使用随机数生成器,因为我需要按照规则进行计算和预测。但是,您可以使用 PyTorch 中的随机数生成器生成随机自然数。以下是一个示例代码:
```
import torch
# 生成一个在 [0, 100) 范围内的随机整数
random_int = torch.randint(low=0, high=100, size=(1,)).item()
print(random_int)
```
该代码使用 `torch.randint()` 函数生成一个在 [0, 100) 范围内的随机整数,并使用 `.item()` 方法将其转换为 Python 中的整数类型。您可以根据需要更改 `low` 和 `high` 参数以生成不同范围内的随机整数。
阅读全文