Skip to main contentModule nll
Source - CrossEntropyLossBackward
- Cross-entropy loss backward.
- CrossEntropyLossForward
- Cross-entropy loss forward:
out[n] = log(sum_c exp(x[n,c])) - x[n, target[n]]. - MultilabelSoftMarginLossBackward
- Multi-label soft-margin loss backward (element-wise).
- MultilabelSoftMarginLossForward
- Multi-label soft-margin loss forward (element-wise).
- NllLossBackward
- NLL loss backward:
dx[n, target[n]] = -dy[n], zero elsewhere. - NllLossForward
- NLL loss forward:
out[n] = -log_prob[n, target[n]].
- cross_entropy_loss_backward
- Cross-entropy loss backward.
- cross_entropy_loss_forward
- Cross-entropy loss forward:
out[n] = log(sum_c exp(x[n,c])) - x[n, target[n]]. - multilabel_soft_margin_loss_backward
- Multi-label soft-margin loss backward (element-wise).
- multilabel_soft_margin_loss_forward
- Multi-label soft-margin loss forward (element-wise).
- nll_loss_backward
- NLL loss backward:
dx[n, target[n]] = -dy[n], zero elsewhere. - nll_loss_forward
- NLL loss forward:
out[n] = -log_prob[n, target[n]].