mindspore.numpy.triu_indices_from

查看源文件
mindspore.numpy.triu_indices_from(arr, k=0)[源代码]

返回 arr 上三角的索引。

参数:
  • arr (Union[Tensor, list, tuple]) - 二维数组。

  • k (int, 可选) - 对角线的偏移量,默认值: 0

返回:

triu_indices_from ,由两个Tensor组成的的Tuple, arr 的上三角形的索引。

异常:
  • TypeError - 如果 arr 不能转换为一个Tensor,或者 k 不是一个数字。

  • ValueError - 如果 arr 不能转换为一个二维的Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.numpy as np
>>> tensor = np.ones((3,3))
>>> print(np.triu_indices_from(tensor))
(Tensor(shape=[6], dtype=Int32, value= [0, 0, 0, 1, 1, 2]),
Tensor(shape=[6], dtype=Int32, value= [0, 1, 2, 1, 2, 2]))