Argmax performance slower than numpy?

3

Thanks, filed: https://github.com/pytorch/pytorch/issues/8817