bi#dZddlmZmZddlZddlmcmZ ddl mcm Z ddl mZddlmZddlmZddlmZGddeZy) a, Copyright 2013 Steven Diamond Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. )ListTupleN)AffAtom) Constraint) is_param_freeceZdZdZdfd Zej dZddZde e e ffdZ de fdZ de fdZde e e ffd Zde fd Zde fd Zd Zd Z dde e dfde ej*eeffdZxZS)kronzKronecker product. returnc.tt| ||yN)superr __init__)selflh_exprrh_expr __class__s R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/affine/kron.pyrz kron.__init__!s dD"7G4c:tj|d|dS)z-Kronecker product of the two values. r)npr )rvaluess rnumericz kron.numeric$swwvay&),,rc|jdjs(|jdjs td|jdjdk7s|jdjdk7r tdy)zKChecks that both arguments are vectors, and the first is constant. rrz/At least one argument to kron must be constant.z'kron requires both arguments to be 2-d.N)args is_constant ValueErrorndimrs rvalidate_argumentszkron.validate_arguments*ss ! ((*diil.F.F.HNO O YYq\  ! #tyy|'8'8A'=FG G(>rc|jdjd|jdjdz}|jdjd|jdjdz}||fS)Nrr)rshape)rrowscolss rshape_from_argszkron.shape_from_args2slyy|!!!$TYYq\%7%7%::yy|!!!$TYYq\%7%7%::TzrcjtjjrZ|jd}|jd}|j xs|j xrt |xr t |S|jdj xs|jdj S)zIs the atom convex? rr)uscopesdpp_scope_activerrr)rxys ris_atom_convexzkron.is_atom_convex7s 88 $ $ & ! A ! A]]_7 <"1%:-*: =99Q<++-K11I1I1K Krc"|jS)zIs the atom concave? )r-r s ris_atom_concavezkron.is_atom_concaveCs""$$rcvtjj|jd|jdS)zSame as times. rr)r(signmul_signrr s rsign_from_argszkron.sign_from_argsHs)vvtyy|TYYq\::rc~|jdjrdnd}|j|jS)z;Is the composition non-decreasing in argument idx? rr)rr is_nonnegridxcst_locs ris_incrz kron.is_incrM6yy|//1!qyy!++--rc~|jdjrdnd}|j|jS)z;Is the composition non-increasing in argument idx? rr)rr is_nonposr6s ris_decrz kron.is_decrSr:rc|jdjxr|jdj}|jdjxr|jdj}|xs|S)zCheck a *sufficient condition* that the expression is PSD, by checking if both arguments are PSD or both are NSD. rrris_psdis_nsdrcase1case2s rr@z kron.is_psdYf ! ##%?$))A,*=*=*? ! ##%?$))A,*=*=*?~rc|jdjxr|jdj}|jdjxr|jdj}|xs|S)zCheck a *sufficient condition* that the expression is NSD, by checking if one argument is PSD and the other is NSD. rrr?rBs rrAz kron.is_nsdarErr#.c|jdjrtj|d|d|gfStj|d|d|gfS)aKronecker product of two matrices. Parameters ---------- arg_objs : list LinOp for each argument. shape : tuple The shape of the resulting expression. data : Additional data required by the atom. Returns ------- tuple (LinOp for objective, list of constraints) rr)rrlukron_rkron_l)rarg_objsr#datas rgraph_implementationzkron.graph_implementationisX& 99Q< # # %99Xa[(1+u=rA A99Xa[(1+u=rA Ar)r Nr )__name__ __module__ __qualname____doc__rr numpy_numericrr!rintr&boolr-r/r3r9r=r@rAloLinOprrrM __classcell__)rs@rr r s5 -- HsCx L L%% ;dDj 1; .d. .d. 6:B$S#XB rxxj)) *Brr )rQtypingrrnumpyrcvxpy.lin_ops.lin_oplin_opslin_oprUcvxpy.lin_ops.lin_utils lin_utilsrHcvxpy.utilities utilitiesr(cvxpy.atoms.affine.affine_atomrcvxpy.constraints.constraintr%cvxpy.expressions.constants.parameterrr rrres:!!$$23?cB7cBr