Diagflat

Creates a diagonal matrix with the flattened input as the diagonal.

Abstract Signature:

Diagflat(input: Tensor, offset: int = 0)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.diagflat

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.numpy.diagflat

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.diagflat

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.ops.numpy.diagflat

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.linalg.diag

Strategy: Macro 'tf.linalg.diag(tf.reshape({v}, [-1]), k={k})'

Apple MLX

API: —

Strategy: Custom / Partial

Flax NNX

API: jax.numpy.diagflat

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: jax.numpy.diagflat

Strategy: Direct Mapping

Official Docs ↗