torch.range(start, end)和torch.arange(start, end)
举例:
import torch a = torch.range(1, 6) print('a: ', a) b = torch.arange(1, 6) print('b: ', b)
a: tensor([1., 2., 3., 4., 5., 6.])
b: tensor([1, 2, 3, 4, 5])
从输出可以明显看出,range包括end,而arange不包括end。同时可以看出两者的数据类型也不同,如下所示。
print('type(a): ', a.dtype) print('type(b): ', b.dtype)
type(a): torch.float32
type(b): torch.int64