For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo function

leaky_relu

def leaky_relu[dtype: DType, simd_width: SIMDSize](x: SIMD[dtype, simd_width], negative_slope: Scalar[dtype]) -> SIMD[dtype, simd_width]

Compute the Leaky ReLU using the equation $max(x, 0) + negative_slope * min(x, 0)$ .

Constraints:

Type must be a floating point Dtype.

Parameters:

dtype (DType): DType used for the computation.
simd_width (SIMDSize): SIMD width used for the computation.

Args:

x (SIMD[dtype, simd_width]): The value to compute the Leaky ReLU operation on.
negative_slope (Scalar[dtype]): The slope for negative values.

Returns:

SIMD[dtype, simd_width]: The result of the Leaky ReLU operation.