Structs§
- BceLoss
Backward - Element-wise BCE backward.
- BceLoss
Forward - Element-wise binary cross-entropy forward.
- BceWith
Logits Loss Backward - Element-wise BCE-with-logits backward.
- BceWith
Logits Loss Forward - Element-wise BCE-with-logits forward.
- Gaussian
NllLoss Backward Input - Element-wise Gaussian NLL backward w.r.t. input (mean prediction).
- Gaussian
NllLoss Forward - Element-wise Gaussian NLL loss forward.
- KlDiv
Loss Backward - Element-wise KL-divergence backward w.r.t. log-input.
- KlDiv
Loss Forward - Element-wise KL-divergence forward.
- Poisson
NllLoss Backward - Element-wise Poisson NLL loss backward (log_input=True).
- Poisson
NllLoss 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.
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.