Hi there, I am trying to get index from dataloader, which loads from torch.sparse.FloatTensor.
The code is like:
sparseTensor = torch.sparse.FloatTensor(i, v) train_data = DataLoader(sparseTensor) for input in train_data: # Need the index of the input here.
The input is a sparse matrix, and I need the row number for each row trained.
Thanks for your time!