Tri

An array with ones at and below the given diagonal and zeros elsewhere.

Abstract Signature:

Tri(N: int, M: int, k: int = 0, dtype: dtype)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Macro 'torch.tril(torch.ones(({N}, {M} if {M} is not None else {N}), dtype={dtype}), diagonal={k})'

JAX (Core)

API: jax.numpy.tri

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.tri

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.ops.tri

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.experimental.numpy.tri

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.core.tri

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: jax.numpy.tri

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: jax.numpy.tri

Strategy: Direct Mapping

Official Docs ↗