LeakyRelu

LeakyRelu takes input data (Tensor<T>) and an argument alpha, and produces one output data (Tensor<T>) where the function f(x) = alpha * x for x < 0, f(x) = x for x >= 0, is applied to the data tensor elementwise.

Abstract Signature:

LeakyRelu(X: Tensor, alpha: float)

PyTorchKerasTensorFlowApple MLX

PyTorch

API: torch.nn.modules.activation.LeakyReLU

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.layers.LeakyReLU

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: keras.layers.LeakyReLU

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.layers.activations.LeakyReLU

Strategy: Direct Mapping

Official Docs ↗