nonzero isn't consistent with sort in terms of dimensions (though both return an indices vector).

>>> torch.rand(10).nonzero().size()
(10L, 1L)
>>> torch.rand(10).sort()[1].size()
(10L,)