nn_multilabel_soft_margin_loss {torch} | R Documentation |
Multi label soft margin loss
Description
Creates a criterion that optimizes a multi-label one-versus-all
loss based on max-entropy, between input and target
of size
.
Usage
nn_multilabel_soft_margin_loss(weight = NULL, reduction = "mean")
Arguments
weight |
(Tensor, optional): a manual rescaling weight given to each
class. If given, it has to be a Tensor of size |
reduction |
(string, optional): Specifies the reduction to apply to the output:
|
Details
For each sample in the minibatch:
where ,
.
Shape
Input:
where
N
is the batch size andC
is the number of classes.Target:
, label targets padded by -1 ensuring same shape as the input.
Output: scalar. If
reduction
is'none'
, then.