mlx_rs::nn

Function hard_swish

Source
pub fn hard_swish(x: impl AsRef<Array>) -> Result<Array>
Expand description

Applies the hardswish function, element-wise.

This is:

x * minimum(maximum(x + 3, 0), 6) / 6