How to make torch.nonzero faster

6

found right timing method here: Measuring GPU tensor operation speed - #4 by apaszke