biVdZddlmZddlZddlmZddlm Z ddl m Z Gdde 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. )TupleN)AtomceZdZdZdfd Zdfd ZdZdZdee dffdZ dee e ffd Z de fd Z de fd Zde fd Zde fd Zde fdZdZxZS) sum_largestz9 Sum of the largest k values in the expression X returnc:||_tt||yN)ksuperr__init__)selfxr __class__s R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/sum_largest.pyr zsum_largest.__init__s k4)!,c^|jdkr tdtt|y)z,Verify that k is a positive number. rz*Second argument must be a positive number.N)r ValueErrorr rvalidate_arguments)r rs rrzsum_largest.validate_arguments#s) 66Q;IJ J k435rc|dj}t|}ttj|j }|j |z }|dkDr||kr9tj | |d|dz}||j} ||| zz }|S)z Returns the sum of the k largest entries of the matrix. For non-integer k, uses linear interpolation. rkthNg) flattenlenintnpfloorr argpartitionminsum) r valuesvaluenk_floork_fracindicesresult indices_next next_values rnumericzsum_largest.numeric*s q !!# Jbhhtvv&''! Q;oouf#gq/A2EFxPG7^'')FF A:'A+??E6w? 1ML|,002J fz) )F rctj|djj}t |}t t j|j}|j|z }t j|jdjd|jdjdzdf}|dkDr-t j| t||dz d|}d||<|dkDrC||kr>t j| |d|dz}|t j||} ||| <tj |gS)a+Gives the (sub/super)gradient of the atom w.r.t. each argument. Matrix expressions are vectorized, so the gradient is a matrix. Args: values: A list of numeric values for the arguments. Returns: A list of SciPy CSC sparse matrices or None. rrrN)intf from_2D_to_1DrTrrrrr zerosargsshaperrargminsp csc_array) r r!r"r#r$r%Dr&r(next_idxs r_gradzsum_largest._gradEs#""6!9#4#4#6#8#89 Jbhhtvv&''! HHdiil((+DIIaL,>,>q,AA1E F Q;oouf#gq/A2EFxPGAgJ A:'A+??E6w? 1ML#BIIeL.A$BCH AhK Q  r.ctS)z8Returns the (row, col) shape of the expression. )tupler s rshape_from_argszsum_largest.shape_from_argses wrcv|jdj|jdjfS)zCReturns sign (is positive, is negative) of the expression. r)r0 is_nonneg is_nonposr:s rsign_from_argszsum_largest.sign_from_argsjs1 ! &&($))A,*@*@*BCCrcy)zIs the atom convex? Tr:s ris_atom_convexzsum_largest.is_atom_convexprcy)zIs the atom concave? FrAr:s ris_atom_concavezsum_largest.is_atom_concaveurcy)z;Is the composition non-decreasing in argument idx? TrAr idxs ris_incrzsum_largest.is_incrzrCrcy)z;Is the composition non-increasing in argument idx? FrArHs ris_decrzsum_largest.is_decrrFrc:td|jDS)z&Is the atom piecewise linear? c3<K|]}|jywr )is_pwl).0args r z%sum_largest.is_pwl..s5C3::<5s)allr0r:s rrOzsum_largest.is_pwls5499555rc|jgS)z!Returns the parameter k. )r r:s rget_datazsum_largest.get_datasxr)rN)__name__ __module__ __qualname____doc__r rr*r7rrr;boolr?rBrErJrLrOrU __classcell__)rs@rrrs-66!@sCx DdDj 1D   d d 66 rr)rYtypingrnumpyr scipy.sparsesparser3cvxpy.interface interfacer,cvxpy.atoms.atomrrrArrrcs) !r$rr