如何在Pytorch中形成一个连续的数字序列？

Question

如何转换Matlab代码

v = [1: n]

到 pytorch？

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

Answer 1

你在 python

中形成了一个连续的数字序列

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

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

torch_v = torch.from_numpy(v)

Answer 2

您可以直接使用 Pytorch 中的 arange 方法。

torch_v = torch.arange(1,n)

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