jaxrts.experimental.SiiNN.NNModelExpandedZ.set_norms

NNModelExpandedZ.set_norms(T: float, rho: float, Z: list[float], k: float)

Set the normalization of the input layers.