FillDiagonal

Fill the main diagonal of the given array of any dimensionality.

Abstract Signature:

FillDiagonal(a: Array, val, wrap: bool = False)

PyTorch

API: torch.Tensor.fill_diagonal_
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.fill_diagonal
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.fill_diagonal
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.linalg.set_diag
Strategy: Direct Mapping
Official Docs ↗