mlx.nn.HardShrink

目录

mlx.nn.HardShrink#

class HardShrink#

应用 HardShrink 函数。

有关函数式等价物,请参阅 hard_shrink()

参数:

lambd – Hardshrink 的 \(\lambda\) 值。默认值: 0.5

方法