nonzero isn't consistent with sort in terms of dimensions (though both return an indices vector). >>> torch.rand(10).nonzero().size() (10L, ...
確定! 回上一頁