ott.solvers.nn.models.ICNN.act_fn

ICNN.act_fn(x): Callable[[Array], Array] = <jax._src.custom_derivatives.custom_jvp object>

Parameters:

x (Any) –

Return type:

Any