pytorch使用（四）np.random.randint用法

np.random.randint 用法

np.random.randint 是 numpy 库中用于生成随机整数的函数。它的用法如下：

numpy.random.randint(low, high=None, size=None, dtype='l')
其中，各个参数的含义如下：

low：生成的随机整数的下限（包含）。
high：生成的随机整数的上限（不包含）。如果不提供 high 参数，则生成的随机整数的上限为 low，下限为 0。
size：指定生成的随机整数的形状。可以是整数，元组或列表。
dtype：指定生成的随机整数的数据类型。默认为长整型（'l'）。
下面是一些使用示例：

import numpy as np

# 生成一个范围在 [0, 5) 之间的随机整数
print(np.random.randint(5))

# 生成一个范围在 [2, 6) 之间的随机整数
print(np.random.randint(2, 6))

# 生成一个 3x3 的随机整数矩阵，范围在 [0, 10) 之间
print(np.random.randint(10, size=(3, 3)))

# 生成一个 2x3x4 的随机整数矩阵，范围在 [-5, 5) 之间
print(np.random.randint(-5, 5, size=(2, 3, 4)))

需要注意的是，生成的随机整数不包括上限。如果需要包括上限，可以将上限加 1，然后在生成随机整数后再减去 1。