Structs§
- Leaky
Relu Backward - Backward: dx = dy if x > 0 else negative_slope * dy
- Leaky
Relu Forward - Forward: y = x if x > 0 else negative_slope * x
- Leaky
Relu Forward Dispatch - Forward: y = x if x > 0 else negative_slope * x
- Leaky
Relu Op - Softplus
Backward - Backward: dx = dy * sigmoid(betax) For betax > threshold: dx ≈ dy
- Softplus
Forward - Forward: y = (1/beta) * log(1 + exp(betax)) For betax > threshold: y ≈ x (numerically safe pass-through)
- Softplus
Forward Dispatch - Forward: y = (1/beta) * log(1 + exp(betax)) For betax > threshold: y ≈ x (numerically safe pass-through)
- Softplus
Op - Softshrink
Backward - Backward: dx = dy if |x| > lambda else 0
- Softshrink
Forward - Forward: y = x - lambda if x > lambda, x + lambda if x < -lambda, else 0
- Softshrink
Forward Dispatch - Forward: y = x - lambda if x > lambda, x + lambda if x < -lambda, else 0
- Softshrink
Op - Softsign
Backward - Backward: dx = dy / (1 + |x|)²
- Softsign
Forward - Forward: y = x / (1 + |x|)
- Softsign
Forward Dispatch - Forward: y = x / (1 + |x|)
- Softsign
Op - Threshold
Backward - Backward: dx = dy if x > threshold else 0
- Threshold
Forward - Forward: y = x if x > threshold else value
- Threshold
Forward Dispatch - Forward: y = x if x > threshold else value
- Threshold
Op
Functions§
- leaky_
relu_ backward - Backward: dx = dy if x > 0 else negative_slope * dy
- leaky_
relu_ forward - Forward: y = x if x > 0 else negative_slope * x
- softplus_
backward - Backward: dx = dy * sigmoid(betax) For betax > threshold: dx ≈ dy
- softplus_
forward - Forward: y = (1/beta) * log(1 + exp(betax)) For betax > threshold: y ≈ x (numerically safe pass-through)
- softshrink_
backward - Backward: dx = dy if |x| > lambda else 0
- softshrink_
forward - Forward: y = x - lambda if x > lambda, x + lambda if x < -lambda, else 0
- softsign_
backward - Backward: dx = dy / (1 + |x|)²
- softsign_
forward - Forward: y = x / (1 + |x|)
- threshold_
backward - Backward: dx = dy if x > threshold else 0
- threshold_
forward - Forward: y = x if x > threshold else value