bihdZddlmZddlZddlmZddlm Z ddl m Z Gdde Z d de d e fd Zy) a, Copyright 2018 Akshay Agrawal 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)hstack)AxisAtomceZdZdZddeddffd ZdeeeffdZdefdZdefdZ defd Z defd Z defd Z defd Z d ZdZdZxZS)ProdaMultiply the entries of an expression. The semantics of this atom are the same as np.prod. This atom is log-log affine, but it is neither convex nor concave. Parameters ---------- expr : Expression The expression to multiply the entries of. axis : int The axis along which to sum. keepdims : bool Whether to drop dimensions after summing. Nkeepdimsreturnc2tt| |||y)Naxisr)superr__init__)selfexprr r __class__s K/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/prod.pyrz Prod.__init__+s dD"4dX"Fc@|jdjryy)zCReturns sign (is positive, is negative) of the expression. r)TF)FFargs is_nonnegrs rsign_from_argszProd.sign_from_args.s 99Q< ! ! # rcy)zIs the atom convex? Frs ris_atom_convexzProd.is_atom_convex5rcy)zIs the atom concave? Frrs ris_atom_concavezProd.is_atom_concave:rrcy)z$Is the atom log-log convex? Trrs ris_atom_log_log_convexzProd.is_atom_log_log_convex?rcy)z%Is the atom log-log concave? Trrs ris_atom_log_log_concavezProd.is_atom_log_log_concaveDr"rc<|jdjS)z;Is the composition non-decreasing in argument idx? rrridxs ris_incrz Prod.is_incrIsyy|%%''rcy)z;Is the composition non-increasing in argument idx? Frr&s ris_decrz Prod.is_decrNrrctj|dr|d}|jq|j|jd|jdzk(r |j }n!t jd|j}t j|}|S|jdvr|j|j|j|jk(}t j|jd|jz |j}|jdk(r |dd|fn||ddf}t j|j|j||<|jr-t j||j}|Std|St j|d|j|j}|S) z3Takes the product of the entries of value. rN)dtype)rr,)r z6cp.prod does not support axis > 1 for sparse matrices.r )intf is_sparser nnzshapedatanpzerosr-prod_getnnztoarrayr expand_dims UserWarning)rvaluessp_matr2resultmasks rnumericz Prod.numericSso >>&) $AYFyy ::a6<<?!BB!;;D88AV\\:D f$~~499~5dii9PP&,,q{";6<<P*.))q.vagfT1Wo!wwt||~DIIFt ==^^FDII>F  ""Z[[ WWVAYTYYOF rc2tj||z S)a Gives the (sub/super)gradient of the atom w.r.t. a column argument. Matrix expressions are vectorized, so the gradient is a matrix. Args: value: A numeric value for a column. Returns: A NumPy ndarray or None. )r3r5)rvalues r _column_gradzProd._column_gradnswwu~%%rc$|j|S)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. ) _axis_grad)rr:s r_gradz Prod._grad{sv&&rNF)__name__ __module__ __qualname____doc__boolrrrrrr!r$r(r*r>rArD __classcell__)rs@rrrs G$G4GdDj 1    (d( d 6 & 'rrrr cdt|trtt|St|||S)a9Multiply the entries of an expression. The semantics of this atom are the same as np.prod. This atom is log-log affine, but it is neither convex nor concave. Parameters ---------- expr : Expression or list[Expression, Numeric] The expression to multiply the entries of, or a list of Expressions and numeric types. axis : int The axis along which to take the product; ignored if `expr` is a list. keepdims : bool Whether to drop dimensions after taking the product; ignored if `expr` is a list. ) isinstancelistrr)rr rs rr5r5s,$$F4L!!D$))rrE)rItypingrnumpyr3cvxpy.interface interfacer.cvxpy.atoms.affine.hstackrcvxpy.atoms.axis_atomrrrJr5rrrrUs= ,*l'8l'^*D*T*r