GroupNormalization¶

Group normalization layer.

Abstract Signature:

GroupNormalization(groups: int = 32, axis: int = -1, epsilon: float = 0.001, center: bool = True, scale: bool = True)

API: torch.nn.GroupNorm

Strategy: Direct Mapping

API: keras.layers.GroupNormalization

Strategy: Direct Mapping

API: tf.keras.layers.GroupNormalization

Strategy: Direct Mapping

API: mlx.nn.GroupNorm

Strategy: Direct Mapping

API: flax.nnx.GroupNorm

Strategy: Direct Mapping