mlx.nn.Softshrink

目錄

mlx.nn.Softshrink#

class Softshrink(lambd=0.5)#

Applies the Softshrink function.

See softshrink() for the functional equivalent.

參數:

lambd -- the \(\lambda\) value for Softshrink. Default: 0.5

方法