mlx.nn.hard_tanh

目录

mlx.nn.hard_tanh#

class hard_tanh(x, min_val=-1.0, max_val=1.0)#

应用 HardTanh 函数。

逐元素应用 \(\max(\min(x, \text{max\_val}), \text{min\_val})\)