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


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


用法

matvec(
    x, adjoint=False, name='matvec'
)

参数

  • x Tensor 形状兼容,dtypeself 相同。 x 被视为 [batch] 向量,意思是每组前导维度,最后一个维度定义一个向量。有关兼容性的定义,请参见类文档字符串。
  • adjoint Pythonbool。如果 True ,左乘伴随:A^H x
  • name Op 的名称。

返回

  • 形状为 [..., M]Tensor 和与 self 相同的 dtype

用左乘法变换 [batch] 向量xx --> Ax

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

X = ... # shape [..., N], batch vector

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

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

相关用法


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