当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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 维张量返回。

offsetlower 参数确定要提取的三角形:

  • 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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。