a Ÿ¬d3d4„Z?d5d6„Z@d7d8„ZAd9d:„ZBd;d<„ZCd=d>„ZDd?d@„ZEdAdB„ZFdCdD„ZGdEdF„ZHdGdH„ZIdIdJ„ZJdKdL„ZKdMdN„ZLdOdP„ZM‡ZNS)RÚTensorflowPrinterz„ Tensorflow printer which handles vectorized piecewise functions, logical operators, max/min, and relational operators. Z_tensorflowcodeztensorflow.math.absztensorflow.math.signztensorflow.math.ceilztensorflow.math.floorztensorflow.math.logztensorflow.math.expútensorflow.math.sqrtztensorflow.math.cosztensorflow.math.acosztensorflow.math.sinztensorflow.math.asinztensorflow.math.tanztensorflow.math.atanztensorflow.math.atan2ztensorflow.math.coshztensorflow.math.acoshztensorflow.math.sinhztensorflow.math.asinhztensorflow.math.tanhztensorflow.math.atanhztensorflow.math.realztensorflow.math.imagztensorflow.math.angleztensorflow.math.erfztensorflow.math.lgammaztensorflow.math.equalztensorflow.math.not_equalztensorflow.math.greaterztensorflow.math.lessztensorflow.math.less_equalztensorflow.math.greater_equalztensorflow.math.logical_andztensorflow.math.logical_orztensorflow.math.logical_notztensorflow.math.maximumztensorflow.math.minimumútensorflow.math.addztensorflow.math.multiplyztensorflow.linalg.traceztensorflow.linalg.detN)Útensorflow_versioncs2tƒ |¡|jd}|dur(tr(tj}||_dS)Nr)ÚsuperÚ__init__Z _settingsr Ú __version__r)ÚselfÚsettingsÚversion©Ú __class__©úi/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/printing/tensorflow.pyrQs    zTensorflowPrinter.__init__cslˆj t|ƒd¡}|dur&tƒ |¡S‡fdd„|jDƒ}t|ƒdkr\dˆ |¡|dfSˆ ||¡SdS)Ncsg|]}ˆ |¡‘qSr©Ú_print©Ú.0Úarg©rrrÚ ]óz5TensorflowPrinter._print_Function..éú%s(%s)r) ÚmappingÚgetÚtyperZ _print_BasicÚargsÚlenÚ_module_formatÚ_expand_fold_binary_op)rÚexprÚopÚchildrenrrrÚ_print_FunctionYs  þz!TensorflowPrinter._print_FunctioncCs| d¡}d || |j¡¡S)Nztensorflow.linalg.invú{}({}))r(Úformatrr)rr*r+rrrÚ_print_Inversers z TensorflowPrinter._print_InversecCsD|j}|r&t|ƒtdƒkr&| d¡}n | d¡}d || |j¡¡S)Nz1.14ztensorflow.matrix_transposez"tensorflow.linalg.matrix_transposer.)rrr(r/rr)rr*rr+rrrÚ_print_Transposevs   z"TensorflowPrinter._print_Transposecs:|j}tdd„|Dƒƒr tdƒ‚‡‡fdd„‰ˆ|j|ƒS)Ncss|]}t|tƒVqdS©N)Ú isinstancer©rÚirrrÚ €r z6TensorflowPrinter._print_Derivative..z1derivation by multiple variables is not supportedcs:|sˆ |¡Sdˆ d¡ˆ||dd…ƒˆ |d¡fS)Nz %s(%s, %s)[0]ztensorflow.gradientséÿÿÿÿ)rr()r*r&©rÚunfoldrrr9‚s  ýz3TensorflowPrinter._print_Derivative..unfold)Ú variablesÚanyÚNotImplementedErrorr*)rr*r:rr8rÚ_print_Derivative~s z#TensorflowPrinter._print_Derivativec Cs¦|j}|r t|ƒtdƒkr d}nd}ddlm}|jdj\}}t|jƒdkrpd | |¡| |¡| |¡d¡Sd | |¡| |¡| |¡| ||jdd…Ž¡¡S)Nz1.0ztensorflow.selectztensorflow.wherer)Ú Piecewiser!z{}({}, {}, {})) rrZ$sympy.functions.elementary.piecewiser>r&r'r/r(r)rr*rZtensorflow_piecewiser>ÚeZcondrrrÚ_print_PiecewiseŒs& üüz"TensorflowPrinter._print_PiecewisecCsN|j\}}|jtjkr.d | d¡| |¡¡Sd | d¡| |¡| |¡¡S)Nr.r z {}({}, {})ztensorflow.math.pow)r&ÚexprZHalfr/r(r)rr*ÚbaserArrrÚ _print_Pow¢s  ÿþzTensorflowPrinter._print_PowcsD|jr dnd}dd ‡fdd„| ¡Dƒ¡d}dˆ |¡|fS) Nztensorflow.Variableztensorflow.constantú[ú, cs,g|]$}dd ‡fdd„|Dƒ¡d‘qS)rDrEcsg|]}ˆ |¡‘qSrr©rÚjrrrr¯r zBTensorflowPrinter._print_MatrixBase...ú])Újoinr4rrrr¯r z7TensorflowPrinter._print_MatrixBase..rHr")Z free_symbolsrIÚtolistr()rr*Z tensorflow_fÚdatarrrÚ_print_MatrixBase­s $þz#TensorflowPrinter._print_MatrixBasecsnddlm‰‡fdd„|jDƒ‰‡fdd„|jDƒ}|r^d| t |¡td¡| dˆ¡fS| dˆ¡SdS) Nr©Ú MatrixExprcsg|]}t|ˆƒr|‘qSr)r3rrMrrr·r z3TensorflowPrinter._print_MatMul..csg|]}|ˆvr|‘qSrrr)Úmat_argsrrr¸r z%s*%srútensorflow.linalg.matmul)Zsympy.matrices.expressionsrNr&Z parenthesizerZfromiterrr))rr*r&r)rNrOrÚ _print_MatMulµs ÿþÿzTensorflowPrinter._print_MatMulcCs| d|jg|j¡S)NrP)r)rBrA©rr*rrrÚ _print_MatPowÃsÿzTensorflowPrinter._print_MatPowcCsd| |j¡| |j¡fS)Nz%s = %s)rÚlhsÚrhsrRrrrÚ_print_AssignmentÇs  þz#TensorflowPrinter._print_AssignmentcCs*g}|jD]}| | |¡¡q d |¡S)NÚ )r&ÚappendrrI)rr*ÚretZsubexprrrrÚ_print_CodeBlockÎs z"TensorflowPrinter._print_CodeBlockccs@tddƒD]}t|ƒVq tddƒD]}t|ƒVq$tdƒ‚dS)Néaé{éAé[zout of letters)ÚrangeÚchrÚ ValueError)rr5rrrÚ _get_letter_generator_for_einsumÕs   z2TensorflowPrinter._get_letter_generator_for_einsumcsLˆ ¡‰d ‡fdd„|jDƒ¡}dˆ d¡|d ‡fdd„|jDƒ¡fS)Nú,cs(g|] }d ‡fdd„t|ƒDƒ¡‘qS)Úcsg|] }tˆƒ‘qSr)ÚnextrF©ÚlettersrrrÞr zJTensorflowPrinter._print_ArrayTensorProduct...)rIr_r4rfrrrÞr z?TensorflowPrinter._print_ArrayTensorProduct..ú %s("%s", %s)útensorflow.linalg.einsumrEcsg|]}ˆ |¡‘qSrrrrrrrâr )rbrIÚsubranksr(r&)rr*Úcontraction_stringr)rgrrÚ_print_ArrayTensorProductÜsýz+TensorflowPrinter._print_ArrayTensorProductc s|ddlm}|j}|j}ˆ |j|¡\}}}|s:ˆ |¡St||ƒrr‡fdd„|jDƒ}dˆ  d¡|d  |¡fSt ƒ‚dS)Nr©ÚArrayTensorProductcsg|]}dˆ |¡‘qS)z%srrrrrrîr z=TensorflowPrinter._print_ArrayContraction..rhrirE) Ú0sympy.tensor.array.expressions.array_expressionsrnr*Úcontraction_indicesÚ_get_einsum_stringrjrr3r&r(rIr<) rr*rnrBrprkÚ letters_freeÚ letters_dumÚelemsrrrÚ_print_ArrayContractionås   ýz)TensorflowPrinter._print_ArrayContractionc sºddlm}t|jƒ}t|ƒdkr&t‚t|dƒdkr:t‚t|j|ƒrX|jj}|jj }n|j}|jg}ˆ  ||¡\}}}‡fdd„|Dƒ}dˆ  d¡d   |d   |¡d   |¡¡d   |¡fS) Nrrmr!écsg|]}ˆ |¡‘qSrrr4rrrrr z:TensorflowPrinter._print_ArrayDiagonal..rhriz{}->{}{}rdrE)rornÚlistÚdiagonal_indicesr'r<r3r*rjr&rqr(r/rI) rr*rnrxrjrtZdiagonal_stringrrrsrrrÚ_print_ArrayDiagonalös$     ýz&TensorflowPrinter._print_ArrayDiagonalcCs&d| d¡| |j¡| |jj¡fS)Nz %s(%s, %s)ztensorflow.transpose)r(rr*Z permutationZ array_formrRrrrÚ_print_PermuteDimss   ýz$TensorflowPrinter._print_PermuteDimscCs| d|j¡S)Nr )r)r&rRrrrÚ_print_ArrayAddsz!TensorflowPrinter._print_ArrayAdd)N)OÚ__name__Ú __module__Ú __qualname__Ú__doc__Z printmethodÚsympyZAbsÚsignZceilingÚfloorÚlogrArÚcosÚacosÚsinÚasinÚtanÚatanÚatan2ÚcoshÚacoshÚsinhÚasinhÚtanhÚatanhÚreZimrÚerfZloggammaÚEqZNeZStrictGreaterThanZStrictLessThanZLessThanZ GreaterThanÚAndÚOrÚNotZMaxZMinZMatAddZHadamardProductZTraceZ Determinantr#Údictr Z_default_settingsrr-Z _print_ExprZ_print_ApplicationZ_print_MatrixExprZ_print_RelationalZ _print_NotZ _print_AndZ _print_OrZ_print_HadamardProductZ _print_TraceZ_print_Determinantr0r1r=r@rCrLrQrSrVrZrbrlruryrzr{Ú __classcell__rrrrr s–Ì7þ   r cKst|ƒ}| |¡Sr2)r Zdoprint)r*rÚprinterrrrÚtensorflow_codesrš)Zsympy.external.importtoolsrÚcollections.abcrZsympy.core.mulrZsympy.core.singletonrZsympy.codegen.cfunctionsrZsympy.externalrZsympy.printing.precedencerZsympy.printing.pycoder r€r r ršrrrrÚs