mindspore.numpy.diagflat ================================= .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/br_base/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/br_base/docs/api/api_python/numpy/mindspore.numpy.diagflat.rst :alt: 查看源文件 .. 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。