I have a large tensor of size [80000, 300]. I want to select a set of n rows from this 2D tensor. So the resultant tensor should have a shape [n, 300]. I looked at this 2 similar topic, and it proposed a very simple solution of using torch.ByteTensor. However my problem is I can not initialise that large ByteTensor.
I tried create a ByteTensor for indexing the main 2D tensor as follows. I first create a zero tensor of size 80000, then I fill the positions of required rows with 1’s. For example, If I want to select the 1st, 5th, and 7th rows, I’d do the following.
ind = torch.zeros(80000)
ind[0] = 1
ind[4] = 1
ind[6] = 1

Then how do I obtain the ByteTensor out of this tensor? Simply doing torch.ByteTensor(ind) is raising error.