ott.neural.networks.icnn.ICNN.act_fn# ICNN.act_fn(x): Callable[[Array], Array] = <jax._src.custom_derivatives.custom_jvp object># Parameters: x (Array | ndarray | bool | number | bool | int | float | complex) Return type: Array