randint関数【torch】

説明：整数の一様分布に従う乱数

書式：

torch.randint( 下限値、上限値、形状 )

code:torch_randint1.py

import torch as pt

print(type(t1))

print(t1.shape)

print(t1)

形状はリストやタプルで与える。