linear_operator_udvh_update.py 文件源码

def _apply(self, x, adjoint=False):
    u = self.u
    v = self.v
    l = self.base_operator
    d = self.diag_operator

    leading_term = l.apply(x, adjoint=adjoint)

    if adjoint:
      uh_x = math_ops.matmul(u, x, adjoint_a=True)
      d_uh_x = d.apply(uh_x, adjoint=adjoint)
      v_d_uh_x = math_ops.matmul(v, d_uh_x)
      return leading_term + v_d_uh_x
    else:
      vh_x = math_ops.matmul(v, x, adjoint_a=True)
      d_vh_x = d.apply(vh_x, adjoint=adjoint)
      u_d_vh_x = math_ops.matmul(u, d_vh_x)
      return leading_term + u_d_vh_x