bi-VdZddlmZddlZddlmZddlm Z ddl m Z Gdde Z y)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 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)Atom)cvxtypesceZdZdZdfd ZdZdefdZdfd Zde e dffdZ d Z de e e ffd Zde fd Zde fd Zd Zde fdZde fdZde fdZde fdZddZxZS)gmatmulaGeometric matrix multiplication; :math:`A \mathbin{\diamond} X`. For :math:`A \in \mathbf{R}^{m \times n}` and :math:`X \in \mathbf{R}^{n \times p}_{++}`, this atom represents .. math:: \left[\begin{array}{ccc} \prod_{j=1}^n X_{j1}^{A_{1j}} & \cdots & \prod_{j=1}^n X_{pj}^{A_{1j}} \\ \vdots & & \vdots \\ \prod_{j=1}^n X_{j1}^{A_{mj}} & \cdots & \prod_{j=1}^n X_{pj}^{A_{mj}} \end{array}\right] This atom is log-log affine (in :math:`X`). Parameters ---------- A : cvxpy.Expression A constant matrix. X : cvxpy.Expression A positive matrix. returnc`tj||_tt||yN)r cast_to_constAsuperr__init__)selfr X __class__s N/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/cvxpy/atoms/gmatmul.pyrzgmatmul.__init__1s&##A& gt%a(ctj|d}tj|jj|zS)z)Geometric matrix multiplication. r)nplogexpr value)rvalueslogXs rnumericzgmatmul.numeric8s2vvfQi vvdffllT)**rcn|jjd|jd|jddS)N(z, r))r__name__r argsrs rnamez gmatmul.name>s+#~~66#vv#yy|- -rc^tt| |jj s t d|jj r3t|jtjs t d|jdjs t dy)z6Raises an error if the arguments are invalid. z*gmatmul(A, X) requires that A be constant.z;gmatmul(A, X) requires that A be a Constant or a Parameter.rz*gmatmul(A, X) requires that X be positive.N) r rvalidate_argumentsr is_constant ValueError parameters isinstancer parameterr is_pos)rrs rr$zgmatmul.validate_argumentsCs gt/1vv!!#<  66   z$&&(:L:L:N'OM yy|""$< %r.ctjj|jj|jdjS)z8Returns the (row, col) shape of the expression. r)ushape mul_shapesr r r!s rshape_from_argszgmatmul.shape_from_argsTs1ww!!$&&,, ! 0B0BCCrc|jgS)zLReturns info needed to reconstruct the expression besides the args. )r r!s rget_datazgmatmul.get_dataYsxrcy)zCReturns sign (is positive, is negative) of the expression. )TFr!s rsign_from_argszgmatmul.sign_from_args^srcy)zIs the atom convex? Fr3r!s ris_atom_convexzgmatmul.is_atom_convexcrcy)zIs the atom concave? Fr3r!s ris_atom_concavezgmatmul.is_atom_concavehr7rcr|jdj|jjzS)Nr)r r'r r!s rr'zgmatmul.parametersms+yy|&&(466+<+<+>>>rctjjr>|jd}|j}|j xr|j Sy)z$Is the atom log-log convex? rT)r,scopesdpp_scope_activer r r')rrr s ris_atom_log_log_convexzgmatmul.is_atom_log_log_convexqsI 88 $ $ & ! AA 91<<>: :rc"|jS)z%Is the atom log-log concave? )r>r!s ris_atom_log_log_concavezgmatmul.is_atom_log_log_concaves**,,rc6|jjS)z;Is the composition non-decreasing in argument idx? )r is_nonnegridxs ris_incrzgmatmul.is_incrvv!!rc6|jjS)z;Is the composition non-increasing in argument idx? )r is_nonposrCs ris_decrzgmatmul.is_decrrFrcyr r3)rrs r_gradz gmatmul._gradsr)rN)r __module__ __qualname____doc__rrstrr"r$rintr/r1boolr4r6r9r'r>r@rErIrK __classcell__)rs@rrrs,)+ -c- "DsCxD  dDj 1   ?,-- "d" "d" rr) rNtypingrnumpyrcvxpy.utilities utilitiesr,cvxpy.atoms.atomrcvxpy.expressionsrrr3rrrYs) !&}d}r