\;jrddlZddlZddlmZddlmZddlmZmZm Z m Z ddl m Z ddl mZGdd e Zy) N) unique_name)_append_activation_in_dygraph) Parameter _global_flagsdtype_is_floatingin_dygraph_mode)LayerHelperBase) ParamAttrceZdZfdZdZddZddZedZedZ dZ ddZ dd Z d Z dd Zd Zd ZdZxZS) LayerHelperc ||_|jjdd}|"tj||jd<t||jd|y)Nname) layer_type)kwargsgetrgeneratesuper__init__)selfrrr __class__s aG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/base/layer_helper.pyrzLayerHelper.__init__ sY {{vt, <"-"6"6z"BDKK  V,DcV|jjj|i|SN) main_program current_block append_op)rargsrs rrzLayerHelper.append_op+s):t  ..0::DKFKKrc|jj|g}g}t|ttfr)|D]"}|j |j |$|S|j |j ||Sr)rr isinstancelisttupleappend to_variable)rinput_param_nameinputsretinps rmultiple_inputzLayerHelper.multiple_input.sp!126 ftUm , 4++C01  JJt''/ 0 rch|j|}t|dk7r|jd|dS)Nrz layer only takes one inputr)r*lenr)rr&r's rinputzLayerHelper.input8s<$$%56 v;! __%%@A Aayrc`tj|jjddS)N param_attrr _to_attrrrrs rr/zLayerHelper.param_attr>s"!!$++//,"EFFrc`tj|jjddS)N bias_attrr0r2s rr4zLayerHelper.bias_attrBs"!!$++//+t"DEErc|j}t|tr|g}t|dk7rt||k7r t dt|dk(r8|dk7r3dg|z}t |D]}t j|d||<|}|S)Nrzparameter number mismatchr)r/r!r r, ValueErrorrangecopydeepcopy)rlengthr/tmpis rmultiple_param_attrzLayerHelper.multiple_param_attrGs__ j) ,$J z?a C Ov$=89 9 _ !fk&6/C6]z!}5A#Jrc#K|j|}|jt|}t||Ed{y7wr)r*r=r,zip)rr&r' param_attrss riter_inputs_and_paramsz"LayerHelper.iter_inputs_and_paramsUs;$$%56..s6{; v{+++s;AAAc|j|}d}|D]:}| |j}||jk7s"td||jfz|S)NzData Type mismatch: %d to %d)r*dtyper6)rr&r'rCeachs r input_dtypezLayerHelper.input_dtypeZsa$$%56D} $**$ 2eTZZ5HH  rc|jjj|}t|tst d|z|S)Nzno Parameter name %s found)r global_blockvarr!rr6)rrparams r get_parameterzLayerHelper.get_parameterfsA!!..044T:%+9D@A A rc t|j||}|j}|s|S|j|||jd}|j |j}|j d|g|gdd|gid|i|S) a: Append bias operator and return its output. If the user does not set bias_attr, append_bias_op will return input_var :param input_var: the input variable. The len(input_var.shape) is larger or equal than 2. :bias_initializer: an instance of a subclass of Initializer used to initialize the bias :param dim_start: :param dim_end: the shape of the bias will be input_var.shape[dim_start:dim_end]. The bias is broadcasted to other dimensions and added to input_var to get the output T)attrshaperCis_biasrCelementwise_add)XYOutaxistyper'outputsattrs)r"rMr4create_parameterrC"create_variable_for_type_inferencer)r input_var dim_startdim_endsizer4br;s rappend_bias_opzLayerHelper.append_bias_opmsIOOIg67NN    ! !$ioot " 55IOO5L "#A3/SEN9%   rcn|jjdd}||St|trd|i}nt t|dzd}d|jvr;|jjdr |jjd}||d<|jjdt jdd}|r||d<|j d}trt||||}|S|j|j}|j|d |gid |gi| |S) NactrVz should be unicode or str use_cudnn use_mkldnnFLAGS_use_mkldnnFrOrQrSrU) rrr!str TypeErrorrpopr rrZrCr)rr[rbrcrdact_typeresr;s rappend_activationzLayerHelper.append_activations0kkooeT* ;  c3 3-CCH'BBC C $++ %$++//+*F  4I(C [[__ -/--.@%H   *C 776?  /8Y CJ99 9PC NNi[)   Jrc| t|dur(tjjj Stjjj S)NT)rpaddlenn initializer XavierUniformConstant)rrCs r_get_default_initializerz$LayerHelper._get_default_initializersH =-e4<99((668 899((113 3rc|jj|d}t||s"td||j|j y)Nz1The input {0} parameter of method {1} must be {2})rrr!rgr__name__)r param_nameclsrIs r is_instancezLayerHelper.is_instancesF  D1%%C  &r)r-)rN)rt __module__ __qualname__rrr*r-propertyr/r4r=rArErJr`rkrrrw __classcell__)rs@rr r so EL GGFF , @ F4rr )r8rmr dygraph_utilsr frameworkrrrr layer_helper_baser r/r r rrrs3 8 /!a/ar