randint関数(torch)