idx is torch.cuda.LongTensor
idx dimensions are nx3 (where n is the number of indices)
How can I make the following code run faster?
this one works, but it is very slow.
inp contains all the activations, where the first dim is the number of samples in the current mini batch
for row in range(inp.size(0)):
          for x,y,z in idx:
                    inp.data[int(row),int(x),int(y),int(z)] = 1