pytorch随机生成0-100之间的整数50个

可以使用以下代码在PyTorch中随机生成50个0-100之间的整数：

import torch
import random

rand_nums = torch.randint(low=0, high=100, size=(50,))
print(rand_nums)

这将生成一个包含50个随机整数的PyTorch张量对象，并打印出这些整数。

相关问题

使用pytorch随机生成两个数，且这两个数之间的差最少为3

以下是一个实现方式：

import torch

# 设置随机数种子，保证每次运行结果相同
torch.manual_seed(0)

# 生成两个随机数
a = torch.randint(0, 10, (1,))
b = torch.randint(0, 10, (1,))

# 如果两个数之间差小于3，则重新生成
while torch.abs(a - b) < 3:
    a = torch.randint(0, 10, (1,))
    b = torch.randint(0, 10, (1,))

print("a:", a.item(), "b:", b.item())

解释：

- `torch.manual_seed(0)`：设置随机数种子，保证每次运行结果相同。
- `torch.randint(0, 10, (1,))`：生成一个在0到9之间的随机整数。
- `while torch.abs(a - b) < 3`：如果生成的两个数之间的差小于3，则重新生成。
- `a.item()`和`b.item()`：将tensor转换为python中的int类型。

使用pytorch随机生成三个数，其中前两个数之前的差小于3，另一个数与第一个数之间的差大于3

以下是一个实现此要求的示例代码：

import torch

while True:
    a, b, c = torch.randint(0, 10, size=(3,))
    if abs(a - b) < 3 and abs(c - a) > 3:
        break

print(f"a: {a}, b: {b}, c: {c}")

这段代码使用了一个 `while` 循环来不断生成随机数，直到符合要求为止。其中，`torch.randint(0, 10, size=(3,))` 用于生成三个 0 到 9 之间的随机整数。

在每次循环中，我们检查前两个数之间的差是否小于 3，以及第三个数与第一个数之间的差是否大于 3。如果符合要求，就跳出循环并输出三个数的值。