说明 arange(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor 参考文章: Pytorch——torch.arange()函数