projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 7a764fe) | patch
Add support for batched pdist (#12302)
authorErik Brinkman <erk@fb.com>
Thu, 20 Dec 2018 17:35:08 +0000 (09:35 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 20 Dec 2018 17:41:08 +0000 (09:41 -0800)
commit8db44eda013d71ebc5837e45d5ed16b00993a0d2
tree664f796d32b9167b808dc66190c11d0abd0f67d5tree | snapshot
parent7a764fe270ef06f364e6e504db2ce5959660bd8fcommit | diff
Add support for batched pdist (#12302)

Summary:
This updates pdist to work for batched inputs, and updates the
documentation to reflect issues raised.

closes #9406
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12302

Reviewed By: ezyang

Differential Revision: D13528485

Pulled By: erikbrinkman

fbshipit-source-id: 63d93a6e1cc95b483fb58e9ff021758b341cd4de
aten/src/ATen/native/Distance.cpp diff | blob | history
aten/src/ATen/native/cpu/DistanceOpsKernel.cpp diff | blob | history
aten/src/ATen/native/cuda/DistanceKernel.cu diff | blob | history
test/common_utils.py diff | blob | history
test/test_nn.py diff | blob | history
test/test_torch.py diff | blob | history
torch/nn/functional.py diff | blob | history
Domain: Machine Learning / ML Framework;