BatchNormalization¶
Layer that normalizes its inputs.
Abstract Signature:
BatchNormalization(axis: int = -1, momentum: float = 0.99, epsilon: float = 0.001, center: bool = True, scale: bool = True)
JAX (Core)
API:
—Strategy: Custom / Partial
Layer that normalizes its inputs.
Abstract Signature:
BatchNormalization(axis: int = -1, momentum: float = 0.99, epsilon: float = 0.001, center: bool = True, scale: bool = True)
—