frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-02 14:34:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert D17889288: [pytorch][PR] Migrate soft_margin_loss from the TH to Aten (CUDA+CPU)

Browse Source 
Test Plan: revert-hammer

Differential Revision:
D17889288

Original commit changeset: 9ddffe4dbbfa

fbshipit-source-id: 4fc6156ba38834512b2f735ac0d03e34e69b7286
This commit is contained in:
Edward Yang
2019-10-15 14:33:29 -07:00
committed by Facebook Github Bot
parent 5797f5dd27
commit c44e33b578
15 changed files with 302 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/nn/functional.py
View File

@ -2262,7 +2262,7 @@ def soft_margin_loss(input, target, size_average=None, reduce=None, reduction='m
reduction_enum = _Reduction.legacy_get_enum(size_average, reduce)
else:
reduction_enum = _Reduction.get_enum(reduction)
return torch.soft_margin_loss(input, target, reduction_enum)
return torch._C._nn.soft_margin_loss(input, target, reduction_enum)
def multilabel_soft_margin_loss(input, target, weight=None, size_average=None,