NativeGroupNorm

Native group normalization.

Abstract Signature:

NativeGroupNorm(input: Tensor, weight: Tensor, bias: Tensor, N: int, C: int, HxW: int, group: int, eps: float)

PyTorchJAX (Core)Keras

PyTorch

API: torch.native_group_norm

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.nn.group_norm

Strategy: Direct Mapping

Official Docs ↗

Keras

API: —

Strategy: Custom / Partial