Explicit pdist captures (#16286)
Summary:
Per discussion with cpuhrsch
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16286
Differential Revision: D13883001
Pulled By: erikbrinkman
fbshipit-source-id: 86f35d35fde5db67e3fbb09abc418da0116c9aac