torch.arange

torch.arange(start=0, end, step=1)

生成一个按步长递增的序列

import torch
v1 = torch.arange(5)
# tensor([0, 1, 2, 3, 4])
 
v2 = torch.arange(2, 7)
# tensor([2, 3, 4, 5, 6])
 
v3 = torch.arange(1, 4, 0.5)
# tensor([1.0, 1.5, 2.0, 2.5, 3.0, 3.5])

可用于生成位置编码等