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!