fill_diagonal_

paddle.Tensor. fill_diagonal_ ( x, value, offset=0, wrap=False, name=None )

Note

This API is ONLY available in Dygraph mode.

This function fill the value into the x Tensor’s diagonal inplace.

Parameters
  • x (Tensor) – x is the original Tensor

  • value (Scale) – value is the value to filled in x

  • offset (int,optional) – the offset to the main diagonal. Default: 0 (main diagonal).

  • wrap (bool,optional) – the diagonal ‘wrapped’ after N columns for tall matrices.

  • name (str,optional) – Name for the operation (optional, default is None)

Returns

Tensor, Tensor with diagonal filled with value.

Examples

>>> import paddle
>>> x = paddle.ones((4, 3)) * 2
>>> x.fill_diagonal_(1.0)
>>> print(x.tolist())
[[1.0, 2.0, 2.0], [2.0, 1.0, 2.0], [2.0, 2.0, 1.0], [2.0, 2.0, 2.0]]