bifdZddlmZmZddlZddlmZddl m Z ddl m Z ddlmZGdde Zy)a, Copyright, the CVXPY authors 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 https://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)Atom)is_param_affineceZdZdZdfd Zdfd ZdZdZdee dffdZ dee e ffd Z de fd Z de fd Zde fd Zde fd ZdZde fdZedeej,deej,ej,ffdZxZS)dotsorta Computes :math:`\langle sort\left(vec(X)\right), sort\left(vec(W)\right) \rangle`, where :math:`X` is an expression and :math:`W` is constant. | Both arguments are flattened, i.e., we define :math:`x=vec(X)`, :math:`w=vec(W)`. | If the length of :math:`w` is less than the length of :math:`x`, it is conceptually padded with zeros. | When the length of :math:`w` is larger than the length of :math:`x`, an exception is raised. `dotsort` is a generalization of `sum_largest` and `sum_smallest`: | `sum_largest(x, 3)` is equivalent to `dotsort(x,[1,1,1])` | `sum_largest(x, 3.5)` is equivalent to `dotsort(x,[1,1,1,0.5])` | `sum_smallest(x,3)` is equivalent to `-dotsort(x, [-1,-1,-1])` When the constant argument is not a boolean vector, `dotsort` can be considered as a weighted sum of :math:`x`, where the largest weight is assigned to the largest entry in :math:`x`, etc.. returnc.tt| ||y)N)superr__init__)selfXW __class__s N/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/dotsort.pyr zdotsort.__init__/s gt%a+c|jdjs td|jdj|jdjkr tdtt |y)Nz The W argument must be constant.rz8The size of of W must be less or equal to the size of X.)args is_constant ValueErrorsizer rvalidate_arguments)r rs rrzdotsort.validate_arguments2s`yy|'')?@ @ 99Q<  tyy|00 0WX X gt/1rc|j|\}}tj|tj|zS)z Returns the inner product of the sorted values of vec(X) and the sorted (and potentially padded) values of vec(W). )_get_args_from_valuesnpsort)r valuesxw_paddeds rnumericzdotsort.numeric:s4 008 8wwqzBGGH---rc|j|\}}tj|}t|}tj|}t j ||tj|ff|dfgS)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. r)shape)rrargsortlenrsp csc_arrayzeros)r rrr indicesnsorted_ws r_gradz dotsort._gradBsf008 8**Q- F778$ h"((1+(>?1vNOOr.ctS)z8Returns the (row, col) shape of the expression. )tupler s rshape_from_argszdotsort.shape_from_argsUs wrc*|jdj}|jdj}|jdj}|jdj}|xr|xs|xr|}|xr|xs|xr|}||fS)zCReturns sign (is positive, is negative) of the expression. rr)r is_nonneg is_nonpos)r x_posx_negw_posw_neg is_positive is_negatives rsign_from_argszdotsort.sign_from_argsZs ! &&( ! &&( ! &&( ! &&(rds, !Azdzr