torch.randint
function randint(high: number, shape: number[], options?: RandintOptions): Tensorfunction randint(low: number, high: number, shape: number[], options?: RandintOptions): TensorCreate a tensor filled with random integers in [low, high).
Parameters
highnumber- Highest integer (exclusive)
shapenumber[]- Dimensions of the tensor
optionsRandintOptionsoptional- Optional settings for randint
Returns
Tensor– A new tensor with random integersExamples
const a = torch.randint(10, [2, 3]); // Random ints in [0, 10)
const b = torch.randint(5, 10, [2, 3]); // Random ints in [5, 10)