pred = torch.rand((30000, 10))
pred = pred.cuda(6)
ind = pred[:, 5] > 0.9

st = time.time()
pred = pred[ind]
time_pased = time.time() - st
print(time_pased)

The time_pased is about 10ms. Is it normal? Or did I do anything wrong?