biVdZddlmZddlZddlZddlmZmZmZddl Z ddl m cm Zddlm cmZddlmZddlmZddlmZddlmZddlmZmZdd lmZGd d eZ d Z!y) 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. ) annotationsN)ListLiteralTuple)AffAtom)hstack) Constraint)DEFAULT_ORDER_DEPRECATION_MSG Expression)size_from_shapeceZdZdZ d d fd Zed dZddZddZe jdZ ddZ ddZ d Z d dd ZxZS)reshapeat Reshapes the expression. Vectorizes the expression then unvectorizes it into the new shape. The entries are reshaped and stored in column-major order, also known as Fortran order. Parameters ---------- expr : Expression The expression to reshape shape : tuple or int The shape to reshape to order : F(ortran) or C ct|tjr t|f}tj st |dkDr tdtd|Dr|j||j}t||_ |2tjdd}tj |t"d}|dvsJ||_t&t(|W|y)Nz:Expressions of dimension greater than 2 are not supported.c3&K|] }|dk( ywN).0ds U/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/affine/reshape.py z#reshape.__init__..>s&1qBw&s FUNC_NAMErF)rC) isinstancenumbersIntegralints ALLOW_ND_EXPRlen ValueErrorany _infer_shapesizetuple_shaper replacewarningswarn FutureWarningordersuperr__init__)selfexprshaper-reshape_order_warning __class__s rr/zreshape.__init__3s eW-- .ZME3u:>23 3 && &%%eTYY7EEl =$A$I$I+W`$a ! MM/ ?E """  gt%d+cF|jddk(sJdt|dk(r|f}|S|jd}|d|z dk\sJdtj||d|z \}|dk7rt d|d|dt fd |D}|S) NrzOnly one dimension can be -1.rz(Specified dimension must be nonnegative.z"Cannot reshape expression of size z into shape .c30K|] }|dk(rnywrr)rr specified unspecifieds rrz'reshape._infer_shape..XsOAb+i?Os)countr"indexnpdivmodr#r')r2r&unspecified_index remainderr:r;s @@rr%zreshape._infer_shapeJs{{2!#D%DD# u:?GE !& B a"334I> M#M M>%'YYtU1?P;P5Q%R "KA~ 8l5'QRSOOOE r5cy)z$Is the atom log-log convex? Trr0s ris_atom_log_log_convexzreshape.is_atom_log_log_convex[r5cy)z%Is the atom log-log concave? TrrCs ris_atom_log_log_concavezreshape.is_atom_log_log_concave`rEr5c`tj|d|j|jS)zReshape the value. rr-)r>rr2r-)r0valuess rnumericzreshape.numerices#zz&)TZZtzzBBr5c|jdj}t|j}||k(st d|jdy)zMChecks that the new shape has the same number of entries as the old. rzInvalid reshape dimensions r8N)argsr&r r(r#)r0old_lennew_lens rvalidate_argumentszreshape.validate_argumentsksI))A,##!$++.'!48KKA "r5c|jS)z$Returns the shape argument. )r(rCs rshape_from_argszreshape.shape_from_argsus{{r5c2|j|jgS)zLReturns info needed to reconstruct the expression besides the args. )r(r-rCs rget_datazreshape.get_datazs TZZ((r5c(|d}|ddk(rtj||gfStj|}t|dkrtj||gfStj||ddd}tj|gfS)afReshape 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) rr7rNr)lur transposer")r0arg_objsr2dataargresults rgraph_implementationzreshape.graph_implementations&qk 7c>JJsE*B/ /,,s#C5zQ 3.33Ctt5 V,b11r5)N)r2zint | Tuple[int, ...]r-zLiteral['F', 'C', None]returnNone)r2Tuple[int, ...]r&rr]r_)r]bool)r]r^)r]r_)r2r_r]z!Tuple[lo.LinOp, List[Constraint]])__name__ __module__ __qualname____doc__r/ staticmethodr%rDrGr numpy_numericrKrPrRrTr\ __classcell__)r4s@rrr"s(*. ,%,' ,  ,.    CC  ) 6:2.2 *2r5rct|tr,t|jdk(r|S|j dSt|t j st|ttfr'tj|}|j dSt|tr2g}|D]}t|}|j| t|}|Sddt|zz}t|)Nr7rrIzFThe input to deep_flatten must be an Expression, a NumPy array, an intz> or float, or a nested list thereof. Received input of type %s)rr r"r2flattenr>ndarrayrfloat cast_to_constlist deep_flattenappendrtyper#)xyx0x1msgs rrnrns!Z qww<1 H9939' ' Arzz "jS%L&A  $ $Q 'yysy##!T  Bb!B HHRL  1I R LtTUw V WC S/r5)"rd __future__rrr*typingrrrnumpyr>cvxpy.lin_ops.lin_oplin_opslin_oplocvxpy.lin_ops.lin_utils lin_utilsrVcvxpy.settingssettingsr cvxpy.atoms.affine.affine_atomrcvxpy.atoms.affine.hstackrcvxpy.constraints.constraintr cvxpy.expressions.expressionr r cvxpy.utilities.shaper rrnrr5rrsP#''!!$$2,3R1y2gy2xr5