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。
标签:randint,random,整数,生成,pytorch,随机,np From: https://www.cnblogs.com/czyhbo/p/17569862.html