torch.arange(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)
torch.range(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)
共同点:
输出一维tensor,输入一个数为end,输入两个数为start、end,三个数为start、end、step。
区别:
1.torch.arange()从start开始输出 [ e n d − s t a r t s t e p ] [\frac{end-start}{step} ] [step