bi| rdZddlmZmZddlZddlmcmZ ddl mcm Z ddl mZddlmZGddeZy)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) ConstraintceZdZdZdfd ZdeeeffdZejdZ ddZ dee dffdZ defd Zdefd Zdefd Zdefd Z dd ee dfdeej&eeffdZxZS)tracezThe sum of the diagonal entries of a matrix. Parameters ---------- expr : Expression The expression to sum the diagonal of. returnc,tt| |yN)superr__init__)selfexpr __class__s S/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/affine/trace.pyr ztrace.__init__#s eT#D)c|jdjxs|jdj}|jdjxs|jdj }||fS)zfTrace is nonneg (nonpos) if its argument is elementwise nonneg (nonpos) or psd (nsd). r)args is_nonnegis_psd is_nonposis_nsd)rrrs rsign_from_argsztrace.sign_from_args&sgIIaL**,E ! 0C0C0E IIaL**,E ! 0C0C0E )##rc2tj|dS)z#Sums the diagonal entries. r)npr)rvaluess rnumericz trace.numeric/sxxq ""rc|jdj}|jdjdk7s |d|dk7r tdy)z5Checks that the argument is a square matrix. rz-Argument to trace must be a 2-d square array.N)rshapendim ValueError)rr!s rvalidate_argumentsztrace.validate_arguments5sM ! "" 99Q<   !U1Xq%9LM M&:r.ctS)zAlways scalar. )tuplers rshape_from_argsztrace.shape_from_args<s wrcz|jdjxs|jdjS)Nr)ris_real is_hermitianr's rr*z trace.is_realAs/yy|##%D1)B)B)DDrc$|j Sr )r*r's r is_complexztrace.is_complexDs<<>!!rcy)z$Is the atom log-log convex? Tr's ris_atom_log_log_convexztrace.is_atom_log_log_convexGsrcy)z%Is the atom log-log concave? Fr/r's ris_atom_log_log_concaveztrace.is_atom_log_log_concaveLsrr!c6tj|dgfS)aSum the diagonal entries of the linear expression. Parameters ---------- arg_objs : list LinExpr 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) r)lur)rarg_objsr!datas rgraph_implementationztrace.graph_implementationQs&!%r**r)r Nr )__name__ __module__ __qualname____doc__r rboolrr numpy_numericrr$intr(r*r-r0r2loLinOprrr7 __classcell__)rs@rrrs*$dDj 1$ ## NsCx EE"D"  6:+$S#X+ rxxj)) *+rr)r;typingrrnumpyrcvxpy.lin_ops.lin_oplin_opslin_opr?cvxpy.lin_ops.lin_utils lin_utilsr4cvxpy.atoms.affine.affine_atomrcvxpy.constraints.constraintrrr/rrrKs2!!$$23J+GJ+r