bi9dZddlmZmZddlZddlmcmZ ddl mcm Z ddl mZddlmZGddeZdeefd Zd ed efd Zd eedeefdZy)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) ConstraintceZdZdZddfd ZdefdZejdZ de fdZ de fdZ de fdZ de fd Zde fd Zdeed ffd Zd Z ddeed fdeej*eeffdZxZS) transposez}Transpose an expression. For an n-D expression, if axes are given, the order indicates the permutation of axes. returnc:||_tt||yN)axessuperr__init__)selfexprr __class__s W/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/affine/transpose.pyrztranspose.__init__ s  gt%d+c~|j|jddSd|jdd|jdS)Nrz.Tz transpose(z, axes=))r argsrs rnameztranspose.name%sB 99 iil^2& & ! ~WTYYKqA ArcJtj|d|jS)Nrr )nprr )rvaluess rnumericztranspose.numeric,s||F1IDII66rcy)z$Is the atom log-log convex? Trs ris_atom_log_log_convexz transpose.is_atom_log_log_convex0rcy)z%Is the atom log-log concave? Trrs ris_atom_log_log_concavez!transpose.is_atom_log_log_concave5r!rc<|jdjS)z%Is the expression symmetric? r)r is_symmetricrs rr%ztranspose.is_symmetric:yy|((**rc<|jdjS)z*Is the expression skew-symmetric? r)ris_skew_symmetricrs rr(ztranspose.is_skew_symmetric?syy|--//rc<|jdjS)z%Is the expression Hermitian? r)r is_hermitianrs rr*ztranspose.is_hermitianDr&r.ctj|jdjtjg}tj ||j jS)z7Returns the shape of the transpose expression. r)dtyper)remptyrshaper,rr )rarrs rshape_from_argsztranspose.shape_from_argsIsEhhtyy|))">||Cdii0666rc|jgS)z,Returns the axes for transposition. rrs rget_dataztranspose.get_dataOs {rr.cLtj|d|jgfS)aCreate a new variable equal to the argument transposed. 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)lurr )rarg_objsr.datas rgraph_implementationztranspose.graph_implementationTs"& Xa[$))4b99rr )r N)__name__ __module__ __qualname____doc__rstrrr numpy_numericrboolr r#r%r(r*rintr0r2loLinOprrr7 __classcell__)rs@rrrs , BcB 77  +d+ 040 +d+ 7sCx7  6::$S#X: rxxj)) *:rrr ct||S)aYPermute the dimensions of the expression. Alias for transpose with specified axes. Parameters ---------- expr : AffAtom The expression to permute dimensions of. axes : list or tuple of int The new order of the axes. Returns ------- AffAtom A transpose atom with the specified axes. r)r)rr s r permute_dimsrDis" T %%raxis1axis2cttt|j}||||c||<||<t ||S)a5Swap two axes of the expression. Parameters ---------- expr : AffAtom The expression to swap axes of. axis1 : int The first axis to swap. axis2 : int The second axis to swap. Returns ------- AffAtom A transpose atom with the axes swapped. r)listrangelenr.r)rrErFr s rswapaxesrK|sD" c$**o& 'D#E{DKDKe T %%rsource destinationclt|trt|ts tdt|t|k7r t dt |j Dcgc] }||vs| }}tt||D]\}}|j||t||Scc}w)a|Move axes of the expression to new positions. Parameters ---------- expr : AffAtom The expression to move axes of. source : list of int The original positions of the axes to move. destination : list of int The new positions for the moved axes. Returns ------- AffAtom A new transpose atom with the axes moved. z1Source and destination must be lists of integers.z1Source and destination must have the same length.r) isinstancerH TypeErrorrJ ValueErrorrIndimsortedzipinsertr)rrLrMnorderdestsrcs rmoveaxisrZs" fd #:k4+HKLL 6{c+&&LMMdii( <1AVOQ rdsq!!$$23M:M:^&T#Y&&&#&c&*'49'49'r