19 lines
679 B
ReStructuredText
19 lines
679 B
ReStructuredText
mindspore.numpy.diagflat
|
||
=================================
|
||
|
||
.. py:function:: mindspore.numpy.diagflat(v, k=0)
|
||
|
||
返回一个二维数组，其数组输入作为新输出数组的对角线。
|
||
|
||
.. note::
|
||
在GPU上，支持的数据类型是 ``np.float16`` 和 ``np.float32``。
|
||
|
||
参数：
|
||
- **v** (Tensor) - 输入数据，将其平坦化并设置为输出的第 ``k`` 个对角线。
|
||
- **k** (int, 可选) - 需要的对角线；默认值： ``0`` 即主对角线， :math:`k>0` 意味着对角线高于主对角线，反之亦然。
|
||
|
||
返回：
|
||
2-D Tensor。
|
||
|
||
异常：
|
||
- **TypeError** - 如果输入不是一个Tensor。 |