用法:
mxnet.symbol.linalg.makediag(A=None, offset=_Null, name=None, attr=None, out=None, **kwargs)
- A:(
Symbol
) - 对角线条目的张量 - offset:(
int
,
optional
,
default='0'
) - 对角线与主对角线的偏移量。 0 对应于主对角线,负/正值对应于主对角线下方/上方的对角线。 - name:(
string
,
optional.
) - 结果符号的名称。
- A:(
结果符号。
参数:
返回:
返回类型:
构造一个以输入为对角线的方阵。输入是一个张量
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]]]
相关用法
- Python mxnet.symbol.linalg.maketrian用法及代码示例
- Python mxnet.symbol.linalg.extracttrian用法及代码示例
- Python mxnet.symbol.linalg.syevd用法及代码示例
- Python mxnet.symbol.linalg.syrk用法及代码示例
- Python mxnet.symbol.linalg.sumlogdiag用法及代码示例
- Python mxnet.symbol.linalg.gemm用法及代码示例
- Python mxnet.symbol.linalg.potrf用法及代码示例
- Python mxnet.symbol.linalg.inverse用法及代码示例
- Python mxnet.symbol.linalg.extractdiag用法及代码示例
- Python mxnet.symbol.linalg.trsm用法及代码示例
- Python mxnet.symbol.linalg.trmm用法及代码示例
- Python mxnet.symbol.linalg.gemm2用法及代码示例
- Python mxnet.symbol.linalg.det用法及代码示例
- Python mxnet.symbol.linalg.potri用法及代码示例
- Python mxnet.symbol.linalg.slogdet用法及代码示例
- Python mxnet.symbol.linalg.gelqf用法及代码示例
- Python mxnet.symbol.linalg_potrf用法及代码示例
- Python mxnet.symbol.linalg_gelqf用法及代码示例
- Python mxnet.symbol.linalg_extracttrian用法及代码示例
- Python mxnet.symbol.linalg_sumlogdiag用法及代码示例
注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.symbol.linalg.makediag。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。