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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：

注意：