Module elementwise 

Structs

HuberLossBackward

Element-wise Huber loss backward.

HuberLossForward

Element-wise Huber loss forward.

L1LossBackward

Element-wise L1 backward: dx = dy * sign(x - y).

L1LossForward

Element-wise L1 (MAE) loss forward: out = |x - y|.

MseLossBackward

Element-wise MSE backward: dx = 2 * (x - y) * dy.

MseLossForward

Element-wise MSE loss forward: out = (x - y)^2.

SmoothL1LossBackward

Element-wise SmoothL1 backward.

SmoothL1LossForward

Element-wise SmoothL1 (Huber variant) forward.

Functions

huber_loss_backward

Element-wise Huber loss backward.

huber_loss_forward

Element-wise Huber loss forward.

l1_loss_backward

Element-wise L1 backward: dx = dy * sign(x - y).

l1_loss_forward

Element-wise L1 (MAE) loss forward: out = |x - y|.

mse_loss_backward

Element-wise MSE backward: dx = 2 * (x - y) * dy.

mse_loss_forward

Element-wise MSE loss forward: out = (x - y)^2.

smooth_l1_loss_backward

Element-wise SmoothL1 backward.

smooth_l1_loss_forward

Element-wise SmoothL1 (Huber variant) forward.