AVhS\dZddlZddlZddlZddlZddlZddlmZddlm Z ddlm Z ddlm Z ddlm Z ddl mZdd l mZdd l mZdd lmZdd lmZdd lmZddlmZddlmZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$ dNdZ%ejLdda'dZ(dZ)Gd d!Z*Gd"d#Z+Gd$d%Z,d&Z-ejLd'd(d)va.d*Z/d+Z0d,Z1ejLd-d(d)va2d.Z3d/Z4dNd0Z5dOd1Z6ej2d2g3e5d4d5Z7d6Z8ej2d7g3e6d8d9Z9d:Z:d;Z;ej2dZdAZ?dBZdCZ@dDZAdEZBdFZCdGZDdHZEdIZFdJZGdPdKZHdPdLZIdMZJy)Qz#Utility functions for internal use.N)dtypes)flexible_dtypes)indexed_slices)ops) tensor_util) array_opscond)math_ops) np_arrays) np_dtypes)core)nest) tf_exportc"t|g|dSNr)_canonicalize_axesaxisranks X/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/numpy_ops/np_utils.py_canonicalize_axisr(s TFD )! ,,ctttjrfd}nfd}|Dcgc] }|| c}Scc}w)Nc4tdkfdfdS)NrczSNrsrz6_canonicalize_axes....1s $+rcSrr)rsrrz6_canonicalize_axes....1strr rs`rrz$_canonicalize_axes..0s q%|"rc|dkr|zS|Srrrs rrz$_canonicalize_axes..4sqdr) _maybe_static isinstancerTensor)axesr canonicalizerrs ` rrr,sC t $dkk"MCM*. /$-  // /sAc"ttdS)N signature)hasattrinspectrrr_supports_signaturer+9s + &&rc,tj|S)zConverts a native python or numpy type to TF DType. Args: dtype: Could be a python type, a numpy type or a TF DType. Returns: A tensorflow `DType`. )ras_dtypedtypes r _to_tf_typer0=s  rcxt|tjr |jSt j |S)zConverts a native python or TF DType to numpy type. Args: dtype: Could be a python type, a numpy type or a TF DType. Returns: A NumPy `dtype`. )r#rDTypeas_numpy_dtypenpr/r.s r_to_numpy_typer5Is,v||$    %rc8t|tjr |j}t|tj rF|j j}||dk(Stjtj|dStj|S)z9Returns whether `val` is a scalar value or scalar Tensor.r)r#r ndarraydatarr$shapendimsr equalrrr4isscalar)valr:s rr<r<WsoY&&' ((CT[[! IIOOE  aZ ^^INN3/ 33 ;;s rcv|xr6t|dxr(t|jtxr |jS)N__doc__)r)r#r?str)fs r_has_docstringrBes/NGAy !NjC&@NQYYrc8|jdr|dzS|dzS)N z )endswith)ss r_add_blank_linerGks!ZZ t8O v:rc ttdsy|yt|tjs tj |Sd}|d|j}|d|j}gd}g}||Dcgc]0}tj|tjj2c}z }|jdkDrA||Dcgc]2}tj|tjjd4c}z }|tjdtjj|jdk(rdnd |jzgz }||Dcgc]5\}}tj|tjj|7c}}z }tj|S#t $rYywxYwcc}wcc}wcc}}w) z:An enhanced inspect.signature that can handle numpy.ufunc.r(Ncz|dkrgS|dk(r|gSt|Dcgc]}|t|dzzc}Scc}w)Nr)ranger@)prefixnis rnames_from_numz%_np_signature..names_from_numsAAv i aXo+08 4afs1q5z! 44 4s8xout))whereT)casting same_kind)orderK)r/N)subokT)r(N)extobjNrJ)defaultr)r)r*r#r4ufuncr( ValueErrorninnout ParameterPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORD KEYWORD_ONLY Signature)rArO input_names output_nameskeyword_only_paramsparamsnamerYs r _np_signaturerhrs + & Y  Arxx    q !!5sAEE*+qvv., &  g//??@ &VVaZ    $ 1 1 A A4PF       1 1&&A+$7QVV+; & .  $g//<.relaxs>W   . .0A0A0N0N OO    4 44 Hrr)abrls r_is_compatible_param_kindros qU1X rc|t|tsJ|t|trJ||J ttt|}||J|j }||fS#t$rd}Y#wxYw)aMutually propagates information between `np_fun_name` and `np_fun`. If one is None and the other is not, we'll try to make the former not None in a best effort. Args: np_fun_name: name for the np_fun symbol. At least one of np_fun or np_fun_name shoud be set. np_fun: the numpy function whose docstring will be used. Returns: Processed `np_fun_name` and `np_fun`. N)r#r@getattrr4AttributeError__name__) np_fun_namenp_funs r_prepare_np_fun_name_and_funrvs k3 '' ' &#&& & ^  "" "r3{+,f   //K f  fsA A-,A-c|s|sJ|s |j}d|z}|r!|ddjd|Dzdzz }t|r||jz }t |}t ||||}|S)zHelper to get docs.z%TensorFlow variant of NumPy's `%s`. zUnsupported arguments: z, c3,K|] }d|zdzyw)`Nr).0rgs r z!_np_doc_helper..sDC$J$Dsz. )link)rsjoinrBr?rG _add_np_doc)rAnp_frtunsupported_paramsr|docs r_np_doc_helperrs   --K1K?#! ))D1CD D E  C A199C # C Cd6# *rTF_NP_DOC_FORMstablectS)zvGets the form of the original numpy docstrings. Returns: See `set_np_doc_form` for the list of valid values.  _np_doc_formrrrget_np_doc_formrs rc|ay)aSelects the form of the original numpy docstrings. This function sets a global variable that controls how a tf-numpy symbol's docstring should refer to the original numpy docstring. If `value` is `'inlined'`, the numpy docstring will be verbatim copied into the tf-numpy docstring. Otherwise, a link to the original numpy docstring will be added. Which numpy version the link points to depends on `value`: * `'stable'`: the current stable version; * `'dev'`: the current development version; * pattern `\d+(\.\d+(\.\d+)?)?`: `value` will be treated as a version number, e.g. '1.16'. Args: value: the value to set the global variable to. Nrvalues rset_np_doc_formrs ",rceZdZdZy)Linkc||_yrrselfvs r__init__z Link.__init__ DJrNrs __module__ __qualname__rrrrrr rrceZdZdZy)AliasOfc||_yrrrs rrzAliasOf.__init__rrNrrrrrrrrrc eZdZy)NoLinkN)rsrrrrrrrsrrcv|dk(rd}||zS|dk(rd}||zStjd|r d|d}||zSy) zGenerates link from numpy function name. Args: flag: the flag to control link form. See `set_np_doc_form`. np_fun_name: the numpy function name. Returns: A string. devz;https://numpy.org/devdocs/reference/generated/numpy.%s.htmlrz>https://numpy.org/doc/stable/reference/generated/numpy.%s.htmlz\d+(\.\d+(\.\d+)?)?$zhttps://numpy.org/doc/z"/reference/generated/numpy.%s.htmlN)rematch)flagrttemplates r generate_linkrsk U]LH K  xOH K   xx'.'v-OPH K  rTF_NP_CHECK_LINKFalse)Truetrue1ctSr_is_check_linkrrr is_check_linkr6s rc|ayrrrs rset_check_linkr:s.rc  t}|dk(r4t|r'|d|zz }||jjddz }|St |t r| t ||}nFt |trt ||j}nt |tr |j}nd}|YtrCddl }|j|}|jdk7rtd|d |jd |d |d |d |dz }|S)aAppends the numpy docstring to `doc`, according to `set_np_doc_form`. See `set_np_doc_form` for how it controls the form of the numpy docstring. Args: doc: the docstring to be appended to. np_fun_name: the name of the numpy function. np_f: (optional) the numpy function. link: (optional) which link to use. See `np_doc` for details. Returns: `doc` with numpy docstring appended. inlinedzDocumentation for `numpy.%s`: z>>>>NrzCheck link failed at [z] with status code . Argument `np_fun_name` is .z(See the NumPy documentation for [`numpy.z`](z).)rrBr?replacer#r@rrrrrrequestshead status_coder[)rrtrr|rurlrrs rr~r~?s  $ Yd 0; >>c T\\ ! !% --c2 *1$ | $ ,c D' " $ +c D$  JJc c   MM#  ==C &se+>q}}oN++6-q:    c *rTF_NP_SIG_MISMATCH_IS_ERRORctSr_is_sig_mismatch_an_errorrrris_sig_mismatch_an_errorrus ""rc|ayrrrs rset_is_sig_mismatch_an_errorrys#rcpct\tgfd}|S)aAttachs numpy docstring to a function. Args: np_fun_name: name for the np_fun symbol. At least one of np_fun or np_fun_name shoud be set. np_fun: (optional) the numpy function whose docstring will be used. unsupported_params: (optional) the list of parameters not supported by tf.numpy. link: (optional) which link to use. If `None`, a default link generated from `np_fun_name` will be used. If an instance of `AliasOf`, `link.value` will be used in place of `np_fun_name` for the link generation. If an instance of `Link`, `link.value` will be used as the whole link. If an instance of `NoLink`, no link will be added. Returns: A function decorator that attaches the docstring from `np_fun` to the decorated function. cttdr  tj|}||jj D]@\}} jj |}|Dtr9td|dt jjd d d htrOt|j|js/td|d |jd |jd d d |jtjjk7}|jtjjk7}ts||k7s"td j!||rd nd  jD]"}||jvs j#|$t%| |_|S#t$rd}YwxYw)zThe decorator.r(NzCannot find parameter z@ in the numpy function's signature (which has these parameters: z). Argument `np_fun_name` is z. Argument `np_fun` is rz Parameter z is of kind z while in numpy it is of kind rzbParameter {} should{} have a default value. Argument `np_fun_name` is {}. Argument `np_fun` is {}.z not)rtrr|)r)r*r(r[ parametersitemsgetr TypeErrorlistkeysrokindrYr^emptyformatappendrr?)rAsigrgparamnp_param has_defaultnp_has_defaultr|rurtnp_fun_name_orig np_fun_orignp_sigrs r decoratorznp_doc..decoratorsw $);" >>//1 KD%&&**40(  ')*4&1<&++002345())@ QP % '0Ijj(--1 TF,uzzl;&mm_,H#$$;K=K   ):):)@)@@+#++w/@/@/F/FF. % 'K>,I@@F(Bf$ A / @%% ,D  '  % %d + , -  AI H[sG G$#G$)rvrh)rtrurr|rrrrs```` @@@rnp_docr~sL&#.vK4[&I+v  &2 2 h rc4t\fd}|S)aAttachs numpy docstring to a function. This differs from np_doc in that it doesn't check for a match in signature. Args: np_fun_name: name for the np_fun symbol. At least one of np_fun or np_fun_name shoud be set. np_fun: (optional) the numpy function whose docstring will be used. Returns: A function decorator that attaches the docstring from `np_fun` to the decorated function. c.t||_|S)N)rt)rr?)rArurts rrznp_doc_only..decoratorsq&kBAI Hr)rv)rtrurs`` r np_doc_onlyrs#5[&I+v  rzexperimental.numpy.finfo)v1finfoc>tjt|S)zNote that currently it just forwards to the numpy namesake, while tensorflow and numpy dtypes may have different properties. )r4rr5r.s rrrs .' ((rcnt|tjr|St|tjst j |rt|jSt|tjr |jSt|ttfrtdt|d|S)zFReturns a numpy type if available from x. Skips if x is numpy.ndarray.zJCannot find dtype for type inference from argument `x` of a sequence type zH. For sequences, please call this function on each element individually.)r#numbersRealr IndexedSlicesr is_tf_typer5r/rr2r3rtupler[typerPs r_maybe_get_dtypers7<< H>//0K4J4J14M !'' ""6<<  D%=!  Qy    (rzexperimental.numpy.result_type result_typec tjrtj|\}}|St j |Dcgc] }t |}}|stjgg}tj|Scc}wr) r is_auto_dtype_conversion_enabledrrrflattenrr4asarrayr _result_type)arrays_and_dtypesr/_rPs rrrs ))+**,=>HE1 L#'<<0A#Bq B(   !2 33 sBc|r t|St|trtjSt|t rtj St|S)z5Find the result type from a single input and a dtype.)rr#r@r4str_bytesbytes_)rmr/s rresult_type_unaryrsE u 3 77N!U 99 Qrc tjt|t|S#t$rt ||cYSwxYw)zHA specialization of result_type for 2 arguments for performance reasons.)r rrr[r)t1t2s r_result_type_binaryr)sH  ! !   r2 s'*AAz experimental.numpy.promote_types promote_typesct|}t|}tjtj||Sr)r5r canonicalize_dtyper4r)type1type2s rrr4s6  %  %  % %b&6&6ue&D EErc t|dkr|Stj|d}|ddD]+}tj|tj|}-|Dcgc]}tj||c}Scc}w)zBroadcast tensors. Args: *args: a list of tensors whose shapes are broadcastable against each other. Returns: Tensors broadcasted to the common shape. rJrN)lenrr9broadcast_dynamic_shape broadcast_to)argsshargs r tf_broadcastr<sy Y!^ KtAw" !"XEc  * *2ys/C DBE59 :c) b ) :: :s Bct|tjr-|jjs|jj ryt j|S)aA version of tf.get_static_value that returns None on float dtypes. It returns None on float dtypes in order to avoid breaking gradients. Args: x: a tensor. Returns: Same as `tf.get_static_value`, except that it returns None when `x` has a float dtype. N)r#rr$r/ is_floating is_complexrconstant_valuers rget_static_valuer Qs>4;;QWW%8%8AGGA version of tf.less_equal that eagerly evaluates if possible.rrs r less_equalrrrct|}|2tj|r|r t|S|S|t|zS|t|zS)z?A version of tf.logical_and that eagerly evaluates if possible.r r4r<r"rmrna_values r logical_andr$sQ Q '  {{7 Q }Q' '' }Q rct|}|2tj|r|r|St|S|t|zS|t|zS)z>A version of tf.logical_or that eagerly evaluates if possible.r!r"s r logical_orr&sQ Q '  {{7 Q }Q' '' }Q rct||S)zA version of tf.reduce_all that eagerly evaluates if possible.rkeepdims)r r reduce_allall input_tensorrr,rs rr-r-9|$!Y   |$ JJ 55dX5 ..rcrt|}|tj|||S|j||S)z>A version of tf.reduce_any that eagerly evaluates if possible.r+)r r reduce_anyanyr/s rr3r3r1rc`|jj}||Stj|Sr)r9rr)trs rtf_rankr7s(ggll!] H  r)NNNr)NF)Kr?r*rosrnumpyr4tensorflow.python.frameworkrrrrrtensorflow.python.opsrr rr tensorflow.python.ops.numpy_opsr r tensorflow.python.typesrtensorflow.python.utilrrrrr+r0r5r<rBrGrhrorvrgetenvrrrrrrrrrrr~rrrrrrrrrrrrr r"rrrrrr$r&r)r-r3r7rrrr@s* .76+3+1*55(',- 0'     6#x>>B .ryy)84 (   .-w7;PP . b&BII!7 #$ M`0/B7)8) &5"= ] 4> 4$7B?F@F ;*'"- - - . .    & //r