SpectralNormΒΆ

Spectral normalization.

Abstract Signature:

SpectralNorm(layer, n_steps: int = 1, epsilon: float = 1e-12)

PyTorch

API: torch.nn.utils.spectral_norm
Strategy: Direct Mapping

Keras

API: keras.layers.SpectralNormalization
Strategy: Direct Mapping

TensorFlow

API: tf.keras.layers.SpectralNormalization
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.SpectralNorm
Strategy: Direct Mapping