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


Python tf.linalg.LinearOperatorDiag.matmul用法及代码示例


用法

matmul(
    x, adjoint=False, adjoint_arg=False, name='matmul'
)

参数

  • x LinearOperatorTensor 具有兼容的形状,并且 dtypeself 相同。有关兼容性的定义,请参见类文档字符串。
  • adjoint Pythonbool。如果 True ,左乘伴随:A^H x
  • adjoint_arg Pythonbool。如果 True ,计算 A x^H 其中 x^H 是厄米转置(转置和复共轭)。
  • name Op 的名称。

返回

  • LinearOperatorTensor 形状为 [..., M, R]dtypeself 相同。

用左乘法变换 [batch] 矩阵xx --> Ax

# Make an operator acting like batch matrix A.  Assume A.shape = [..., M, N]
operator = LinearOperator(...)
operator.shape = [..., M, N]

X = ... # shape [..., N, R], batch matrix, R > 0.

Y = operator.matmul(X)
Y.shape
==> [..., M, R]

Y[...,:, r] = sum_j A[...,:, j] X[j, r]

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.linalg.LinearOperatorDiag.matmul。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。