Fix median bug on discontigous tensors (#46917)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46917
fixes https://github.com/pytorch/pytorch/issues/46814
Test Plan: Imported from OSS
Reviewed By: albanD
Differential Revision: D24633412
Pulled By: heitorschueroff
fbshipit-source-id: 54732671b298bdc2b04b13ab3a373892ee0933c3
