bidZddlmZmZddlZddlmZddl m cm Z ddl m cmZddlmZddlmZddlmZddlmZddlmZdd lmZGd d eZGd d eZy)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. )OptionalTupleN)AffAtom)reshape)vec) Constraint) Expression) key_utilsceZdZdZd dfd ZdefdZdefdZdZdZ de e dffd Z de fd Z d d e e dfde ej e effd ZxZS)indexaIndexing/slicing into an Expression. CVXPY supports NumPy-like indexing semantics via the Expression class' overloading of the ``[]`` operator. This is a low-level class constructed by that operator, and it should not be instantiated directly. Parameters ---------- expr : Expression The expression indexed/sliced into. key : The index/slicing key (i.e. expr[key[0],key[1]]) Examples -------- >>> import cvxpy as cp >>> import numpy as np >>> x = cp.Variable((2, 3, 4)) >>> x[..., 2].shape (2, 3) >>> x[..., np.newaxis].shape (2, 3, 4, 1) >>> x[1, 2].shape (4,) returnc|-||_tj||j|_n||_||_t t ||yN) _orig_keyku validate_keyshapekeysuperr __init__)selfexprrorig_key __class__s S/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/affine/index.pyrzindex.__init__:sC   DNsDJJ7DH%DNDH eT#D)cy)zIs the atom log-log convex?Trs ris_atom_log_log_convexzindex.is_atom_log_log_convexDrcy)zIs the atom log-log concave?Trrs ris_atom_log_log_concavezindex.is_atom_log_log_concaveHr!rcddt|jdz zzdz}|jdj|t j |jzzS)z.String representation of the index expression.z[%sz, %s]r)lenrargsnamerto_str)r inner_strs rr)z index.nameLsPFCM!O44s: yy|  "Y4881D%DDDrc&|d|jS)z-Returns the index/slice into the given value.r)rrvaluess rnumericz index.numericQsay((r.ctj|j|j|jdjS)*Returns the shape of the index expression.r)rrrrr(rs rshape_from_argszindex.shape_from_argsUs+xx$..$))A,2D2DEErc2|j|jgS)z&Returns the (row slice, column slice).)rrrs rget_datazindex.get_dataYs$..))rrcDtj|d||d}|gfS) Index/slice into the expression. Parameters ---------- arg_objs : list LinExpr for each argument. shape : tuple The shape of the resulting expression. data : tuple A tuple of slices. r)lur )rarg_objsrdataobjs rgraph_implementationzindex.graph_implementation]s(hhx{E473Ryrr)r N)__name__ __module__ __qualname____doc__rboolr r#r)r/rintr2listr4loLinOprr; __classcell__rs@rr r s4*E )FsCxF*$* 6:$S#X rxxj)) *rr c eZdZdZdeddffd ZdefdZdefdZde fdZ d Z de e d ffd Zdefd Zedeeej(fd Z ddede e d fde ej.eeffdZxZS) special_indexzIndexing using logical indexing or a list of indices. Parameters ---------- expr : Expression The expression being indexed/sliced into. key : tuple ndarrays or lists. rr Nc0||_tj|}tj|j }tj ||jd}|||_|jj|_ tt|3|y)NForder) rr cast_to_constnparangesizerr _select_mat_shaperrHr)rrridx_matrs rrzspecial_index.__init__zsp""4())DII&**Wdjj<"3<&&,,  mT+D1rcy)z$Is the atom log-log convex? Trrs rr z$special_index.is_atom_log_log_convexrcy)z%Is the atom log-log concave? Trrs rr#z%special_index.is_atom_log_log_concaverUrctj|j}|jdj d|dS)z6String representation of the special index expression.r[r&)rspecial_key_to_strrr(r))rkey_strs rr)zspecial_index.names;''1))A,##%&ay22rc&|d|jS)z6Returns the index/slice into the given value. rrr-s rr/zspecial_index.numericsay""r.c|jS)r1)rRrs rr2zspecial_index.shape_from_argss {{rc|jgS)zReturns the key.r\rs rr4zspecial_index.get_dataszrcNtj|j|jjd}t j |j djd}t||t|j ddz|jd}|jS)zGives the (sub/super)gradient of the expression w.r.t. each variable. Matrix expressions are vectorized, so the gradient is a matrix. None indicates variable values unknown or outside domain. rJrKrcscformat) rNrrQrPsp eye_arrayr(rrRgrad)r select_vecidentitylowereds rrezspecial_index.gradsZZ 0 0$2B2B2G2GsS << ! 1 1%@ Z 3tyy|3#? ? KK  ||rr8rc|j}|jj}tj||jd}|d}t j |jdjd}tj||jdjf} ||} tj| | jd} tj| | | jdf} tj| |} | gfS)r6rJrKrr`raT)sparse) rQrrNrrPrcrdr(r7 create_constmul_expr)rr8rr9 select_mat final_shaperfargrgvec_argmul_mat mul_constrlr:s rr;z"special_index.graph_implementations%% &&,, ZZ JOO3G qk<< ! 1 1%@**S499Q<#4#4"67:&OOGW]]4H ;;y'GMM!4D3FGjj;/Ryrr)r<r=r>r?r rr@r r#strr)r/rrAr2rBr4propertyrrc csc_arrayrerCrDrr;rErFs@rrHrHos2Z22  3c3 # sCx$ htBLL12  $"&&*#(c?+0$z:J0J*KrrH)r?typingrrnumpyrN scipy.sparserjrccvxpy.lin_ops.lin_oplin_opslin_oprCcvxpy.lin_ops.lin_utils lin_utilsr7cvxpy.atoms.affine.affine_atomrcvxpy.atoms.affine.reshapercvxpy.atoms.affine.vecrcvxpy.constraints.constraintrcvxpy.expressions.expressionr cvxpy.utilitiesr rr rHrrrrsP#!!$$2.&33+MGM`[G[r