o &i…م@sVddlmZddlmZddlZddlmZdd d „Zddd „ZGdd„dejƒZ dS)é)عannotations)عCallableN)عnnعx1ْtorch.Tensorعx2عreturncCs<t |،rt |،sJdƒ‚t | ،|،St ||،S)zD Complex dot product between tensors x1 and x2: sum(x1.*x2) zx1 and x2 must both be complex)عtorchع is_complexعsumعconj)rr©r ْj/home/dell461/cl/sdc2/last_ska_mid/HISourceFinder-master-l/src/monai/networks/layers/conjugate_gradient.pyع_zdots rعxcCst||ƒ}t |،r |jS|S)z9 Complex dot product between tensor x and itself )rr r عreal)rعresr r rع_zdot_singles rcs6eZdZdZd‡fdd„Zddd„Zddd„Z‡ZS)عConjugateGradientaح Congugate Gradient (CG) solver for linear systems Ax = y. For linear_op that is positive definite and self-adjoint, CG is guaranteed to converge CG is often used to solve linear systems of the form Ax = y, where A is too large to store explicitly, but can be computed via a linear operator. As a result, here we won't set A explicitly as a matrix, but rather as a linear operator. For example, A could be a FFT/IFFT operation ع linear_oprعnum_iterعintcstƒ ،||_||_dS)zs Args: linear_op: Linear operator num_iter: Number of iterations to run CG N)عsuperع__init__rr)عselfrr©ع __class__r rr7s zConjugateGradient.__init__rrعpعrعrsoldrْ=tuple[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor]c Cs`| |،}t||ƒ}||}|||}|||}t|ƒ}||} |}| ||}||||fS)a9 perform one iteration of the CG method. It takes the current solution x, the current search direction p, the current residual r, and the old residual norm rsold as inputs. Then it computes the new solution, search direction, residual, and residual norm, and returns them. )rrr) rrrrrعdyZp_dot_dyعalphaZrsnewعbetar r rعupdateBs zConjugateGradient.updateعycCsT|| |،}t|ƒ}|}t|jƒD]}| ||||،\}}}}|dkr'|Sq|S)a run conjugate gradient for num_iter iterations to solve Ax = y Args: x: tensor (real or complex); Initial guess for linear system Ax = y. The size of x should be applicable to the linear operator. For example, if the linear operator is FFT, then x is HCHW; if the linear operator is a matrix multiplication, then x is a vector y: tensor (real or complex); Measurement. Same size as x Returns: x: Solution to Ax = y g»½×ظك|غ=)rrعrangerr$)rrr%rrrع_ir r rعforwardWs‏zConjugateGradient.forward)rrrr) rrrrrrrrrr )rrr%rrr)ع__name__ع __module__ع__qualname__ع__doc__rr$r(ع __classcell__r r rrr*s r)rrrrrr)rrrr) ع __future__rعtypingrr rrrعModulerr r r rعs