HardShrink

Applies the HardShrink activation function.

Abstract Signature:

HardShrink(x: Tensor, lambd: float = 0.5)

PyTorch

API: torch.nn.functional.hardshrink
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.nn.hard_shrink
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.HardShrink
Strategy: Direct Mapping
Official Docs ↗