Add torch.mode and fix median/kthvalue docs
2 files changed