Step 3: Add support for return_counts to torch.unique for dim not None (#18650)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18650
ghimport-source-id: 75759c95e6c48e27c172b919097dbc40c6bfb5e6
Differential Revision: D14892319
Pulled By: VitalyFedyunin
fbshipit-source-id: ec5d1b80fc879d273ac5a534434fd648468dda1e