pytorch中arange和range的区别

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