函数原型：

arange(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor

返回大小为的一维张量，其值介于区间，以为步长等间隔取值

版权声明：本文为CSDN博主「湘江夜话」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。

原文链接：https://blog.csdn.net/weixin_44504393/article/details/127092330 (opens new window)