tensor笔记6 -- randint

 

输入 

a = torch.randint(0, 6, (4, ))
print(a)

输出：

tensor([0, 2, 4, 4])

 

输入：

b = torch.randint(0, 5, (3))
print(b)

输出：

TypeError: randint() received an invalid combination of arguments - got (int, int, int), but expected one of:
 * (int high, tuple of ints size, *, torch.Generator generator, Tensor out, torch.dtype dtype, torch.layout layout, torch.device device, bool requires_grad)
 * (int low, int high, tuple of ints size, *, torch.Generator generator, Tensor out, torch.dtype dtype, torch.layout layout, torch.device device, bool requires_grad)

 

输入：

c = torch.randint(0, 5, (3, 2))
print(c)

 输出：

tensor([[0, 3],
        [2, 2],
        [2, 2]])

 

参数解释：

前两个参数表示随机取值的的范围；

第三个参数表示返回的tensor形状（x,)表示一维tensor长度为x，逗号不可以省