Threshold

In-place version of Threshold.

Abstract Signature:

Threshold_(input: Tensor, threshold: float, value: float)

PyTorchJAX (Core)NumPyApple MLX

PyTorch

API: torch.threshold_

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro 'jax.numpy.where({input} > {threshold}, {input}, {value})'

NumPy

API: —

Strategy: Macro 'numpy.where({input} > {threshold}, {input}, {value})'

Apple MLX

API: —

Strategy: Macro 'mlx.core.where({input} > {threshold}, {input}, {value})'