forked from mindspore-Ecosystem/mindspore
20 lines
693 B
ReStructuredText
20 lines
693 B
ReStructuredText
|
mindspore.ops.diagflat
|
|||
|
|
======================
|
|||
|
|
|
|||
|
|
.. py:function:: mindspore.ops.diagflat(x, offset=0)
|
|||
|
|
|
|||
|
|
创建一个二维Tensor，用展开后的输入作为它的对角线。
|
|||
|
|
|
|||
|
|
参数：
|
|||
|
|
- **x** (Tensor) - 输入Tensor，展开后作为输出的对角线。
|
|||
|
|
- **offset** (int, 可选) - `offset` 控制选择哪条对角线。默认值：0。
|
|||
|
|
- 当 `offset` 是0时，选择的对角线是主对角线。
|
|||
|
|
- 当 `offset` 大于0时，选择的对角线在主对角线上。
|
|||
|
|
- 当 `offset` 小于0时，选择的对角线在主对角线下。
|
|||
|
|
|
|||
|
|
返回：
|
|||
|
|
二维Tensor。
|
|||
|
|
|
|||
|
|
异常：
|
|||
|
|
- **TypeError** - `x` 不是Tensor。
|