jaxrts.experimental.SiiNN.NNModel.set_norms

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

Set the normalization of the input layers.