How to form a sequence of consecutive numbers in Pytorch?

Question

How to convert the Matlab code

v = [1: n]

to pytorch?

Writing a whole loop for that seems inefficient.

This should be straightforward to find, but I am facing difficulty searching the right keyword to get there.

Answer 1

You can directly use the arange method from Pytorch.

torch_v = torch.arange(1,n)

Reference: https://pytorch.org/docs/master/torch.html?highlight=arange#torch.arange

Answer 2

You form a sequence of consecutive numbers in python

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

if you want a torch tensor you can transform the numpy array like this:

torch_v = torch.from_numpy(v)