pytorch
9c23f6eb - [nn] TripletMarginLoss and PairwiseDistance : no batch dim (#64882)
Commit
View On GitHub
Commit
3 years ago
[nn] TripletMarginLoss and PairwiseDistance : no batch dim (#64882) Summary: Reference: https://github.com/pytorch/pytorch/issues/60585 Pull Request resolved: https://github.com/pytorch/pytorch/pull/64882 Reviewed By: malfet Differential Revision: D31055577 Pulled By: jbschlosser fbshipit-source-id: 2f0a5a08619b672026b48a78bc7d83a6dccba0bf
Author
kshitij12345 kshitij12345
Parents
d35ee431
  • aten/src/ATen/native
    • File
      Distance.cpp
    • File
      Loss.cpp
  • test
    • File
      test_nn.py
  • torch
    • nn/modules
      • File
        distance.py
      • File
        loss.py
    • testing/_internal
      • File
        common_nn.py
FAQ Terms Privacy Refunds Impressum