Skip to main content

Module bce

Module bce 

Source

Structs§

BceLossBackward
Element-wise BCE backward.
BceLossForward
Element-wise binary cross-entropy forward.
BceWithLogitsLossBackward
Element-wise BCE-with-logits backward.
BceWithLogitsLossForward
Element-wise BCE-with-logits forward.
GaussianNllLossBackwardInput
Element-wise Gaussian NLL backward w.r.t. input (mean prediction).
GaussianNllLossForward
Element-wise Gaussian NLL loss forward.
KlDivLossBackward
Element-wise KL-divergence backward w.r.t. log-input.
KlDivLossForward
Element-wise KL-divergence forward.
PoissonNllLossBackward
Element-wise Poisson NLL loss backward (log_input=True).
PoissonNllLossForward
Element-wise Poisson NLL loss forward with log_input=True (default).
SoftMarginLossBackward
Element-wise soft margin loss backward.
SoftMarginLossForward
Element-wise soft margin loss forward.

Functions§

bce_loss_backward
Element-wise BCE backward.
bce_loss_forward
Element-wise binary cross-entropy forward.
bce_with_logits_loss_backward
Element-wise BCE-with-logits backward.
bce_with_logits_loss_forward
Element-wise BCE-with-logits forward.
gaussian_nll_loss_backward_input
Element-wise Gaussian NLL backward w.r.t. input (mean prediction).
gaussian_nll_loss_forward
Element-wise Gaussian NLL loss forward.
kl_div_loss_backward
Element-wise KL-divergence backward w.r.t. log-input.
kl_div_loss_forward
Element-wise KL-divergence forward.
poisson_nll_loss_backward
Element-wise Poisson NLL loss backward (log_input=True).
poisson_nll_loss_forward
Element-wise Poisson NLL loss forward with log_input=True (default).
soft_margin_loss_backward
Element-wise soft margin loss backward.
soft_margin_loss_forward
Element-wise soft margin loss forward.