mindspore.ops.TrilIndices
- class mindspore.ops.TrilIndices(row, col, offset=0, dtype=mstype.int32)[source]
Calculates the indices of the lower triangular elements in a row * col matrix and returns them as a 2-by-N 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 ofmstype.int32
andmstype.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, tril\_size)\), where \(tril\_size\) 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