pytorch
c9bb9907 - [c++] Distance-agnostic triplet margin loss (#45377)

Commit
4 years ago
[c++] Distance-agnostic triplet margin loss (#45377) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45377 This PR adds a C++ implementation of the TripletMarginWithDistanceLoss, for which the Python implementation was introduced in PR #43680. It's based on PR #44072, but I'm resubmitting this to unlink it from Phabricator. Test Plan: Imported from OSS Reviewed By: izdeby Differential Revision: D24003973 fbshipit-source-id: 2d9ada7260a6f27425ff2fdbbf623dad0fb79405
Author
Xinyu Li
Parents
  • test/cpp/api
    • File
      functional.cpp
    • File
      modules.cpp
  • torch
    • csrc/api
      • include/torch/nn
        • functional
          • File
            loss.h
        • modules
          • File
            loss.h
        • options
          • File
            loss.h
      • src/nn/modules
        • File
          loss.cpp
    • nn/modules
      • File
        loss.py
Loading