代码之家  ›  专栏  ›  技术社区  ›  user25004

如何在Pytorch中形成连续的数字序列?

  •  4
  • user25004  · 技术社区  · 8 年前

    如何转换Matlab代码

    v = [1: n]
    

    到Pytork?

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

    2 回复  |  直到 4 年前
        1
  •  15
  •   Avijit Dasgupta    8 年前

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

    torch_v = torch.arange(1,n)
    

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

        2
  •  3
  •   Ioannis Nasios    8 年前

    您可以在python中形成一个连续的数字序列

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

    如果需要火炬张量,可以像这样变换numpy数组:

    torch_v = torch.from_numpy(v)