Python mxnet.ndarray.op.linalg_makediag用法及代码示例

用法:
mxnet.ndarray.op.linalg_makediag(A=None, offset=_Null, out=None, name=None, **kwargs)

参数：

A：(NDArray) - 对角线条目的张量
offset：(int, optional, default='0') - 对角线与主对角线的偏移量。 0 对应于主对角线，负/正值对应于主对角线下方/上方的对角线。
out：(NDArray, optional) - 输出 NDArray 来保存结果。

out：- 此函数的输出。

返回类型：

NDArray 或 NDArray 列表

构造一个以输入为对角线的方阵。输入是一个张量 A 的维度 n >= 1 。

如果 n=1 ，则 A 表示单个方阵的对角线条目。该矩阵将作为二维张量返回。如果 n>1 ，则 A 表示一批方阵的对角线。这批对角矩阵将作为 n+1 维张量返回。

注意：

该运算符仅支持 float32 和 float64 数据类型。

例子：

Single diagonal matrix construction
A = [1.0, 2.0]

makediag(A)    = [[1.0, 0.0],
                  [0.0, 2.0]]

makediag(A, 1) = [[0.0, 1.0, 0.0],
                  [0.0, 0.0, 2.0],
                  [0.0, 0.0, 0.0]]

Batch diagonal matrix construction
A = [[1.0, 2.0],
     [3.0, 4.0]]

makediag(A) = [[[1.0, 0.0],
                [0.0, 2.0]],
               [[3.0, 0.0],
                [0.0, 4.0]]]

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.op.linalg_makediag。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：

注意：