PReLU

Applies the element-wise Parametric ReLU.

Abstract Signature:

PReLU(num_parameters: int = 1, init: float = 0.25)

PyTorchKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.nn.PReLU

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.layers.PReLU

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.keras.layers.PReLU

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.PReLU

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: flax.nnx.PReLU

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: praxis.layers.PReLU

Strategy: Direct Mapping

Official Docs ↗