FillDiagonal

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

Abstract Signature:

FillDiagonal(a: Tensor, val, wrap: bool = False, inplace: bool = True)

PyTorch

API: torch.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 ↗