


a
      tensor([[2., 3., 1.],
              [nan, 1., nan]])
      >>> a.median(0)
      torch.return_types.median(values=tensor([nan, 1., nan]), indices=tensor([1, 1, 1]))
      >>> a.nanmedian(0)
      torch.return_types.nanmedian(values=tensor([2., 1., 1.]), indices=tensor([0, 1, 0]))


