C++ API parity: BCEWithLogitsLoss

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/28783

Test Plan: Imported from OSS

Differential Revision: D18202435

Pulled By: pbelevich

fbshipit-source-id: 011b028bbb2a091e98d3548616b99d7b4569c239
This commit is contained in:
Pavel Belevich
2019-11-20 06:44:36 -08:00
committed by Facebook Github Bot
parent 93db2b86d1
commit f8e7f3fca4
7 changed files with 434 additions and 1 deletions

View File

@ -114,7 +114,7 @@ torch.nn.NLLLoss|Yes|No
torch.nn.PoissonNLLLoss|Yes|No
torch.nn.KLDivLoss|Yes|No
torch.nn.BCELoss|Yes|No
torch.nn.BCEWithLogitsLoss|No|No
torch.nn.BCEWithLogitsLoss|Yes|No
torch.nn.MarginRankingLoss|Yes|No
torch.nn.HingeEmbeddingLoss|Yes|No
torch.nn.MultiLabelMarginLoss|Yes|No