mindspore.ops.TrilIndices

class mindspore.ops.TrilIndices(row, col, offset=0, dtype=mstype.int32)[source]

Computes the indices of the lower triangular elements of a 2D matrix and returns them as a Tensor.

Warning

This is an experimental API that is subject to change or deletion.

Refer to mindspore.ops.tril_indices() for more details.

Parameters

row (int) – number of rows in the 2-D matrix.
col (int) – number of columns in the 2-D matrix.
offset (int, optional) – diagonal offset from the main diagonal. Default: 0 .
dtype (mindspore.dtype, optional) – The specified type of output tensor. An optional data type of mstype.int32 and mstype.int64 . Default: mstype.int32 .

Outputs:

y (Tensor) - indices of the elements in lower triangular part of matrix. The type specified by dtype. The shape of output is $(2, t r i l_s i z e)$ , where $t r i l_s i z e$ is the number of elements in the lower triangular matrix.

Supported Platforms:

Ascend GPU CPU

Examples

>>> from mindspore import ops
>>> from mindspore import dtype as mstype
>>> net = ops.TrilIndices(4, 3, -1, mstype.int64)
>>> output = net()
>>> print(output)
[[1 2 2 3 3 3]
 [0 0 1 0 1 2]]
>>> print(output.dtype)
Int64