Python mxnet.ndarray.op.linalg_extracttrian用法及代碼示例

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

參數：

A：(NDArray) - 方陣張量
offset：(int, optional, default='0') - 對角線與主對角線的偏移量。 0 對應於主對角線，負/正值對應於主對角線下方/上方的對角線。
lower：(boolean, optional, default=1) - 如果 lower=true 則參考下三角矩陣，否則參考上三角矩陣。僅在 offset=0 時相關
out：(NDArray, optional) - 輸出 NDArray 來保存結果。

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

從方陣中提取三角形sub-matrix。輸入是一個張量 A 的維度 n >= 2 。

如果 n=2 ，則 A 表示單個方陣，從中提取三角形 sub-matrix 作為一維張量。

如果 n>2 ，則 A 表示尾隨二維上的一批方陣。提取的三角形 sub-matrices 作為 n-1 維張量返回。

offset 和 lower 參數確定要提取的三角形：

當 offset = 0 根據參數 lower 的值提取相對於主對角線的下三角形或上三角形時。
當offset = k > 0 時，相對於主對角線上方的k-th 對角線的上三角形被提取。
當offset = k < 0 時，提取相對於主對角線下方的k-th 對角線的下三角形。

注意：

該運算符僅支持 float32 和 float64 數據類型。

例子：

Single triagonal extraction
A = [[1.0, 2.0],
     [3.0, 4.0]]

extracttrian(A) = [1.0, 3.0, 4.0]
extracttrian(A, lower=False) = [1.0, 2.0, 4.0]
extracttrian(A, 1) = [2.0]
extracttrian(A, -1) = [3.0]

Batch triagonal extraction
A = [[[1.0, 2.0],
      [3.0, 4.0]],
     [[5.0, 6.0],
      [7.0, 8.0]]]

extracttrian(A) = [[1.0, 3.0, 4.0],
                   [5.0, 7.0, 8.0]]

相關用法

注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.op.linalg_extracttrian。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：

注意：