mindspore.numpy.diagflat

mindspore.numpy.diagflat(v, k=0)

返回一个二维数组，其数组输入作为新输出数组的对角线。

说明

在GPU上，支持的数据类型是 np.float16 和 np.float32。

参数：

• v (Tensor) - 输入数据，将其平坦化并设置为输出的第 k 个对角线。

• k (int, 可选) - 需要的对角线；默认值： 0 即主对角线， \(k>0\) 意味着对角线高于主对角线，反之亦然。

返回：

2-D Tensor。

异常：

• TypeError - 如果输入不是一个Tensor。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> output = np.diagflat(np.asarray([[1,2], [3,4]]))
>>> print(output)
[[1 0 0 0]
[0 2 0 0]
[0 0 3 0]
[0 0 0 4]]
>>> output = np.diagflat(np.asarray([1,2]), 1)
>>> print(output)
[[0 1 0]
[0 0 2]
[0 0 0]]