bi-pdZddlmZmZddlZddlmZddlm Z ddl m Z GddeZ Gd d eZ y) a2 Copyright 2021 the CVXPY developers 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)Cone)cvxtypes)scopesceZdZdZddfd ZdefdZedZdZ de fdZ de fdZ ede fd Zd Zdee fd Zdd e de fd Zdd e de fdZde fdZedee dffdZddZdZxZS) PowCone3DaK An object representing a collection of 3D power cone constraints x[i]**alpha[i] * y[i]**(1-alpha[i]) >= |z[i]| for all i x >= 0, y >= 0 If the parameter alpha is a scalar, it will be promoted to a vector matching the (common) sizes of x, y, z. The numeric value of alpha (or its components, in the vector case) must be a number in the open interval (0, 1). We store flattened representations of the arguments (x, y, z, and alpha) as Expression objects. We construct dual variables with respect to these flattened representations. returncP tj}|j||_|j||_|j||_|j|j|j fD]-}|j r|jr$td|j|}|jra|jjr0tj||jj}ntj|d}||_ tj|jjdks,tj|jjdk\r d}t||jdk(rC|jj|jj|j jdg nV|jj|jj|j j|jjg t fd ddDrdt! z}t|t"t$|O|j|j|j g|y) Nz&All arguments must be affine and real.)rr z=Argument alpha must have entries in the open interval (0, 1).c3.K|] }d|k7ywrNr ).0s arg_shapess R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/constraints/power.py z%PowCone3D.__init__..Ds:az!}!:szIAll arguments must have the same shapes. Provided arguments haveshapes %s)r expression cast_to_constxyz is_affineis_real ValueError is_scalarshapepromotealphanpanyvaluestrsuperr __init__) selfrrrr constr_id Expressionvalmsgr __class__s @rr&zPowCone3D.__init__+s((* ))!,))!,))!,FFDFFDFF+ KCMMO  !IJJ K((/ ?? vv||*((*5$&&,,? +((*5$7 66$**""a' (BFF4::3C3Cq3H,IQCS/ ! ;;$ &&,, dffllBGJ&&,, dffllDJJDTDTUJ ::ab>: :!$Z1CS/ ! i'(@(1 3c pd|jd|jd|jd|jd S)NzPow3D(, ; ))rrrr r's r__str__zPowCone3D.__str__Js!*.&&$&&$&&$**MMr-c tddlm}m}m}m}m}|j j,|jj|jjy||j j}||jj}||jj}t||||jg} ||||||g||j j|jj|jjgz } || | } | jddS)NrMinimizeProblemVariablehstacknorm2SCS:0yE>solvereps)cvxpyr6r7r8r9r:rr#rrrr r solve) r'r6r7r8r9r:rrrconstrobjproblems rresidualzPowCone3D.residualMs ED 66<< 466<<#7466<<;O TVV\\ " TVV\\ " TVV\\ "Aq!TZZ01uVQ1I.#TVV\\466<<$NOPQR#v&}}Et}44r-c2|j|jgSN)r idr2s rget_datazPowCone3D.get_data\s DGG$$r-cyNFr r2s ris_imagzPowCone3D.is_imag_r-cyrKr r2s r is_complexzPowCone3D.is_complexbrMr-c(d|jzSN num_conesr2s rsizezPowCone3D.sizees4>>###r-c.|jjSrG)rrUr2s rrTzPowCone3D.num_conesivv{{r-c*dg|jzSrQrSr2s r cone_sizeszPowCone3D.cone_sizeslss4>>###r-dppc(|ritj5td|jD}t |j t j }|xr|cdddStd|jDS#1swY%xYw)Nc3<K|]}|jywrGrrargs rrz#PowCone3D.is_dcp..rsC#cmmoCc3<K|]}|jywrGr]r^s rrz#PowCone3D.is_dcp..us8s3==?8r`)r dpp_scopeallargs isinstancer r parameterr'rZargs_okexps_oks ris_dcpzPowCone3D.is_dcposz !!# +CCC(X5G5G5IJJ*7 + +8dii888  + +s A BBcyrKr r'rZs ris_dgpzPowCone3D.is_dgpwrMr-c"|jSrGrjr2s ris_dqcpzPowCone3D.is_dqcpz{{}r-.c8d|jjz}|S)N)rR)rr)r'rs rrzPowCone3D.shape}s 466<< r-c tj|d}tj|dddf|jj}tj|dddf|jj}tj|dddf|j j}|j dj||j dj||j dj|y)N)rRrr )r!reshaperrrrdual_variables save_value)r'r#dv0dv1dv2s rsave_dual_valuezPowCone3D.save_dual_values 5'*jjq!tdffll3jjq!tdffll3jjq!tdffll3 A))#. A))#. A))#.r-c|]t|jd|jz |jdd|jz z |jd|jy|Dcgc]}|j}}|jDcgc]}|j}}t |t |jk(sJ||k(sJt|d|jz |dd|jz z |d|jScc}wcc}w)zlImplements the dual cone of PowCone3D See Pg 85 of the MOSEK modelling cookbook for more informationNrr ru)r rwr rrdlen)r'rdr_ args_shapesinstance_args_shapess r _dual_conezPowCone3D._dual_cones < d))!,TZZ79L9LQ9OQRSWS]S]Q]9^))!,djj :1553995K59=#C#CII#C #Ct9DII. .."66 66T!WTZZ/a!DJJ,1G!!Wdjj2 2 6#Cs $DD rGr NF)__name__ __module__ __qualname____doc__r&r$r3propertyrErIboolrLrOintrUrTrrYrjrmrprrr|r __classcell__r,s@rr r s 3>NN 5 5%D$c$$$DI$9$949$4uS#X / 2r-r ceZdZdZdZddeddffd ZdefdZde fdZ de fd Z d Z e d Zd Ze defd ZdeefdZdde de fdZdde de fdZde fdZddZdZxZS) PowConeNDa Represents a collection of N-dimensional power cone constraints that is *mathematically* equivalent to the following code snippet (which makes incorrect use of numpy functions on cvxpy objects): np.prod(np.power(W, alpha), axis=axis) >= np.abs(z), W >= 0 All arguments must be Expression-like, and z must satisfy z.ndim <= 1. The columns (rows) of alpha must sum to 1 when axis=0 (axis=1). Note: unlike PowCone3D, we make no attempt to promote alpha to the appropriate shape. The dimensions of W and alpha must match exactly. Note: Dual variables are not currently implemented for this type of constraint. gư>Naxisr c tj}|j|}|jr|j s d}t ||j|}|j dkDs |jr|j s d}t ||j dkr|jdkDsB|j dk(r|j|jd|z k7s|j dk(r*|dk(r%t d|j|j|fz|j dk(r|j|dkr d}t ||j|}|j|jk7r&t d|jd|jd tj|jd kr t d tjtjdtj|j| z tjkDrt d |z||_||_||_||_|j d k(r|j)d}t*t|[||g|y)Nz2Invalid first argument; W must be affine and real.r zRInvalid second argument. z must be affine, real, and have at most one z.ndim <= 1.ruz>Argument dimensions %s and %s, with axis=%i, are incompatible.z=PowConeND requires left-hand-side to have at least two terms.zArgument dimensions z and z are not equal.rz+Argument alpha must be entry-wise positive.rz+Argument alpha must sum to 1 along axis %s.Forder)rrrrrrndimrUrr!r"r#abssumr_TOL_Wrr rflattenr%r&) r'rrr rr(r)r+r,s rr&zPowConeND.__init__s((*  $ $Q '  FCS/ !  $ $Q ' 66A:aiikakkm7CS/ ! FFaKAFFQJ FFaKAFFaggafo5 FFaKDAIP77AGGT*+, , 66Q;1774=A-QCS/ !((/ ;;!'' ! !67 7 66%++" #JK K 66"&&RVVEKKd;;uVQYYSY%919939;O$PQ#TVV^^#^%>%D%D%)VV^^#^%>%D%D%FGGHI#v&}}Et}44r-c.|jjSrG)rrUr2s rrTzPowConeND.num_conesrWr-czd|jdj|jz}||jzSNr rrdrrrTr' cone_sizes rrUzPowConeND.sizes5 ! **49955 4>>+++r-c|d|jdj|jz}|g|jzSrrrs rrYzPowConeND.cone_sizess7 ! **49955 {T^^---r-rZc2|rtj5|jdjxr|jdj}t |j t j }|xr|cdddSy#1swYyxYw)zCA power cone constraint is DCP if each argument is affine. rr NT)rrbrdrrer rrfrgs rrjzPowConeND.is_dcps !!# +))A,002Otyy|7M7M7O(X5G5G5IJJ*7 + +  +s A+B  BcyrKr rls rrmzPowConeND.is_dgp rMr-c"|jSrGror2s rrpzPowConeND.is_dqcprqr-c@|ddddf}|dddf}|jdk(r|j}|j}|jddk(rtj|}|j dj ||j dj |y)Nrtrr )rTrr!squeezerwrx)r'r#dWdzs rr|zPowConeND.save_dual_values 1crc6] 1b5\ 99>BB 88A;! BB A))"- A))"-r-cl||dk(rL|jd|jz }t||jd|j|jS|Dcgc]}|j}}|j Dcgc]}|j}}t |t |j k(sJ||k(sJ|djj|jjjk(sJt|d|jz |d|j|jScc}wcc}w)zlImplements the dual cone of PowConeND See Pg 85 of the MOSEK modelling cookbook for more informationr rr r)rwr rrrrdr~r#)r'rd scaled_dualsr_rrs rrzPowConeND._dual_cone s  <42:..q1$**+>q+A4::TXT]T]^ ^1553995K59=#C#CII#C #Ct9DII. .."66 667==&&$***:*:*@*@@ @@T!WTZZ/a$**499U U 6#Cs D,;D1rrr)rrrrrrr&r$r3rrLrOrIrrErTrUrrYrjrmrpr|rrrs@rrrs* E#;##;d#;JBBD0 5 5,c,,.DI.$4$4 . Vr-r)rtypingrrnumpyr!cvxpy.constraints.conesrcvxpy.expressionsrcvxpy.utilitiesrr rr r-rrs: (&"A2A2HOVOVr-