mindspore.ops.triu_indices
- mindspore.ops.triu_indices(row, col, offset=0, *, dtype=mstype.int64)[源代码]
计算 row * col 行列矩阵的上三角元素的索引,并将它们作为一个 2xN 的Tensor返回。张量的第一行包含行坐标,第二行包含列坐标。坐标按行排序,然后按列排序。
矩阵的上三角部分包括对角线及其以上的所有元素。
说明
在CUDA上运行的时候, row * col 必须小于2^59以防止计算时溢出。
- 参数:
row (int) - 2-D 矩阵的行数。
col (int) - 2-D 矩阵的列数。
offset (int,可选) - 对角线偏移量。默认值:
0
。
- 关键字参数:
dtype (
mindspore.dtype
,可选) - 指定输出Tensor数据类型,支持的数据类型为mstype.int32
和mstype.int64
,默认值:mstype.int64
。
- 返回:
y (Tensor) - 矩阵的上三角形部分的索引。数据类型由 dtype 指定,shape为 \((2, triu\_size)\) ,其中, \(triu\_size\) 为上三角矩阵的元素总数。
- 异常:
TypeError - 如果 row 、 col 或 offset 不是int。
TypeError - 如果 dtype 的类型不是int32或int64。
ValueError - 如果 row 或者 col 小于零。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> from mindspore import ops >>> output = ops.triu_indices(4, 4, 2, dtype=mindspore.int64) >>> print(output) [[0 0 1] [2 3 3]] >>> print(output.dtype) Int64