mindspore.numpy.tril_indices_from
- mindspore.numpy.tril_indices_from(arr, k=0)[源代码]
返回
arr下三角的索引。- 参数:
arr (Union[Tensor, list, tuple]) - 二维数组。
k (int, 可选) - 对角线的偏移量,默认值:
0。
- 返回:
tril_indices_from,由两个Tensor组成的的Tuple,arr的下三角形的索引。- 异常:
TypeError - 如果
arr不能转换为一个Tensor,或者k不是一个数字。ValueError - 如果
arr不能转换为一个二维的Tensor。
- 支持平台:
AscendGPUCPU
样例:
>>> import mindspore.numpy as np >>> tensor = np.ones((3,3)) >>> print(np.tril_indices_from(tensor)) (Tensor(shape=[6], dtype=Int32, value= [0, 1, 1, 2, 2, 2]), Tensor(shape=[6], dtype=Int32, value= [0, 0, 1, 0, 1, 2]))