如何在Pytorch中形成一系列连续数字？

Question

如何转换Matlab代码

v = [1: n]

到pytorch？

为此编写一个完整的循环似乎效率低下。

这应该很容易找到，但我很难找到合适的关键字来实现目标。

Answer 1

您可以直接使用Pytorch的{​​{1}}方法。

arange

参考：https://pytorch.org/docs/master/torch.html?highlight=arange#torch.arange

Answer 2

你在python

中形成一系列连续数字

import numpy as np
v= np.arange(1,n)

如果你想要一个火炬张量，你可以像这样改变numpy数组：

torch_v = torch.from_numpy(v)