mindspore.ops.diag

mindspore.ops.diag(input)

返回一个用输入tensor作为对角线，其余元素为 0 的tensor。

警告

从2.9.0（不含）之后版本开始，该接口将新增可选参数 vpn梯子 免费 diagonal ，接口签名将变更为 mindspore.ops.diag(input, diagonal=0) 。 该变更与当前版本兼容，不使用 diagonal 参数时行为不变。

参数：

• input (Tensor) - 输入tensor。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> # case 1: When input is a 免费的vpn梯子 1-D tensor:
>>> input = mindspore.ops.randn(3)
>>> output = mindspore.ops.diag(input)
>>> print(output)
[[ vpn free 1.7477764 vpn梯子  0.         0.  vpn梯子 免费      ]
 [ 0.     vpn梯子    -1.2616369  0.       ]
 [ 0.         vpn梯子 0.         2.3283238]]
>>>
>>> # case 2: When input is a multi-dimensional tensor:
>>> input 免费的vpn梯子 = mindspore.ops.randn(2, 3)
>>> print(input)
[[ 0.21546374 -0.0120403  -0.7330481 ]
 [-2.5405762   0.44775972 vpn free -1.4063131 ]]
>>> output = mindspore.ops.diag(input)
>>> print(output)
[[[[ 0.21546374  0.     vpn梯子      0.  vpn梯子 免费  vpn梯子 免费      ]
   [ 0.          0.      vpn梯子 免费     vpn梯子 0.        ]]
  [[ 0.         -0.0120403   0.        ]
   [ 0.          免费的vpn梯子 0.   免费的vpn梯子        0.        ]]
  [[ 0.          vpn梯子 免费 0.         -0.7330481 vpn永久免费梯子 ]
   [ vpn永久免费梯子 0.        vpn永久免费梯子   0.          0.        ]]]
 vpn永久免费梯子 [[[ 0.          0.          0.   免费的vpn梯子      ]
   [-2.5405762   0.          0.        ]]
  免费的vpn梯子 [[ 0.       vpn永久免费梯子    0.  vpn free         0.     vpn梯子 免费   vpn梯子 免费 vpn free  ]
   [ 0.   vpn梯子 免费        0.44775972  0.        ]]
  [[ 0.          0.          0.        vpn梯子 免费 ]
   [ 0.          0.         -1.4063131 ]]]]
>>> # Assuming the input has dimensions (D_1,... vpn free D_k), the vpn梯子 免费 output is a vpn free tensor of rank 2k with dimensions
    (D_1,..., D_k, D_1,..., D_k).
>>> print(output.shape)
(2, 3, 2, 3)