PyTorch torch.randint 函数

torch.randint 是 PyTorch 中用于创建整数随机张量的函数。

torch.randint(low, high, size, dtype, device, requires_grad)

使用示例

import torch

# 创建 0 到 9 的随机整数张量
x = torch.randint(0, 10, (3, 4))

print(x)

输出结果为：

tensor([[4, 5, 2, 6],
        [7, 9, 3, 1],
        [2, 8, 5, 4]])