def multilabel_soft_margin_loss(input, target, weight=None, size_average=True):
"""multilabel_soft_margin_loss(input, target, weight=None, size_average=True) -> Variable
See :class:`~torch.nn.MultiLabelSoftMarginLoss` for details.
"""
input = torch.sigmoid(input)
return binary_cross_entropy(input, target, weight, size_average)
评论列表
文章目录