synax.Residual

class synax.Residual(module: Module)

Residual map.

Computes

\[y = x + f(x)\]

where \(f\) is a given module.

References:

• Deep residual learning for image recognition. 2015. https://arxiv.org/abs/1512.03385.

Parameters:

module – Module to apply.

Methods

init(→ Any)	Sample initial parameters.
parameter_loss(→ jax.Array)	Parameter loss.