AVh|9rdZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd l mZdd l mZdd l mZdd lej$dej$dej$dej$d d;dZdZddZ d;dZdddej@fd Z!ej>d!dej@fd"Z"ej>d#d$ej@fd%Z#ej>d&ej>d'dej@fd(Z$ej>d)dej@fd*Z%ej>d+dej@fd,Z&ej>d-dej@fd.Z'ej>d/dej@fd0Z(ej>d1dej@fd2Z)ej>d3dej@fd4Z*ej>d5ej>d6dej@fd7Z+ej>d8dej@fd9Z,d:Z-y)@z#Ops to manipulate lists of tensors.N) full_type_pb2)cpp_shape_inference_pb2)dtypes)ops)tensor) tensor_shape) tensor_util) array_ops) gen_list_ops)handle_data_util)*TensorListConcatListsTensorListElementShapeTensorListLengthTensorListPushBackBatchcN|d}tjt||||S)N element_shape element_dtypemax_num_elementsname)r empty_tensor_list_build_element_shapers N/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/list_ops.pyrr*s4  ' '(7!'   c 0t|tjrtj|rt j d}n/t|t j st j |}tjj}d|_ |jjtjj|j|jt!j"t j$||_yy)zCSets type information on `list_handle` for consistency with graphs.NT)type_id)shapedtypetype) isinstancer EagerTensorr is_tf_typer TensorShaperCppShapeInferenceResult HandleDatais_setshape_and_typeappendHandleShapeAndTypeas_protoas_datatype_enumr FullTypeDef TFT_ARRAY _handle_data) list_handlerr handle_datas r_set_handle_datar38s S__-m,"..t4m  |'?'? @"..}=m)AALLNKK%%77JJ((*00**=3J3JK K MN +K.rcdtjt||||}t||||S)N)r num_elementsrr)r tensor_list_reserverr3)rr5rrresults rr6r6Ls6  + +(7!   &6=-8 -rctj|}tj|t ||}t ||j |j|S)N)rrr)rconvert_to_tensorr tensor_list_from_tensorrr3rr )rrrr7s rr:r:XsK   (&  / / (7  &66<<6 -rcHtj||t|||S)N) input_handleindexrrr)r tensor_list_get_itemr)r<r=rrrs rr>r>bs)  * * (7!   rc4tj|d||S)Nr)r<rrr)r tensor_list_pop_back)r<rrs rr@r@ls!  * *!   rcHtj||t|||S)N)r<indicesrrr)r tensor_list_gatherr)r<rBrrrs rrCrCts+  ( ((7!   rc tj|}|1tj||||}t j |||Stj ||t|d|}t|||j|S)z?Returns a TensorList created or updated by scattering `tensor`.)r<rrBrr)rrBrr5r) rr9r &tensor_list_scatter_into_existing_listr copy_handle_datatensor_list_scatter_v2rr3r )rrBrr<r output_handles rtensor_list_scatterrIs   (& GG!&'NM%%lMB  77*=9  M ]M6<<@ rcHtj|t||||S)N)r<rrr5r)r tensor_list_stackr)r<rr5rrs rrKrKs+  ' '(7!   rc tj||t|tjgt j |dS)Nr )r<rr leading_dimsrr)r tensor_list_concat_v2rrr9rint64)r<rrrs rtensor_list_concatrQsF  + +!(7((6<<@      rcFtj|t|||S)Nrrlengthsr)r tensor_list_splitrrSs rrUrUs&  ' ' (7   rcftj|||||}tj|||S)z%Sets `item` at `index` in input list.)r<r=itemrresize_if_index_out_of_bounds)r tensor_list_set_itemr rF)r<r=rWrXrrHs rrYrYs; 33   $A -##L-@ rTensorListPushBackopctj|tj|jd|j dS)Nrrr)r r@r rinputsget_attr)r[dresults r _PushBackGradrbs8  * * OOBIIaL1KK0 22rTensorListPopBackc |Gt|jtj|jdt j }|"tj|jd}tj||dfS)Nr shape_typerrr]) rr r tensor_list_element_shapeoutputsrint32r zeros_liketensor_list_push_back)r[dlistdelements r _PopBackGradroso ] nn"<< JJqMfll4 5E##BJJqM2H  + +E8 rr r`rtensor_list_resize) r[rm input_listr=rWrr element_gradinput_list_sizes r_TensorListSetItemGradr$sII*eT// 5y33D9)*%  OOD)JJ , [[%#55jAO// ?KI J ,,rTensorListResizecz|j\}}tj|}tj||dfSN)r_r rr)r[rmrrrs r_TensorListResizeGradr:s7))-*a 33J?/  ( ( @$ FFrTensorListGatherc|j\}}}tj|tj}tj |}t |||j}t|||}|ddfS)z'Gradient function for TensorListGather.re)rrBr<N) r_r rhrrjrr6r rI)r[rsrrBrrr5rms r_TensorListGatherGradrAsm99*gq88V\\+-00<, m\7== I%  gE ;% d rTensorListScatterTensorListScatterV2c  |jd}|jd}tj||tjtj |dgdg|j }|jdk(r|dddfS|ddfS)z(Gradient function for TensorListScatter.rr]rr^rN)r_r rCr r}rr r!)r[rmrrBrss r_TensorListScatterGradrNs 99Q<& IIaL'  + +  OOIOOF$;aS2$GLL  "' WW%% D$ $$ D$ r!TensorListScatterIntoExistingListc |j\}}}tj||tjtj |dgdg|j }tj|}t||||}||dfS)z8Gradient function for TensorListScatterIntoExistingList.r]rr^)r<N) r_r rCr r}rr rkrI)r[rmrrrBrszeross r&_TensorListScatterIntoExistingListGradr_s}yy!VW  + +  OOIOOF$;aS2$GLL  "'   v &% eWgE J%  rczt|tjr|St|tjr|r|j nd}|yt|t jt jfs|s%tj|tjSd}|Dcgc] }|| c}Scc}w)aConverts shape to a format understood by list_ops for element_shape. If `shape` is already a `Tensor` it is returned as-is. We do not perform a type check here. If shape is None or a TensorShape with unknown rank, -1 is returned. If shape is a scalar, an int32 tensor with empty list is returned. Note we do directly return an empty list since ops.convert_to_tensor would conver it to a float32 which is not a valid type for element_shape. If shape is a sequence of dims, None's in the list are replaced with -1. We do not check the dtype of the other dims. Args: shape: Could be None, Tensor, TensorShape or a list of dims (each dim could be a None, scalar or Tensor). Returns: A None-free shape that can be converted to a tensor. NrrMc|yt|tjr|St|tjr|j |j SdS|S)Nr)r" tensor_libTensorr Dimensionr)vals rconvertz%_build_element_shape..convertsL { #z(() j#|--.))/SYY7R7 Jr) r"rrrr%as_listnpndarraygenericrr9rrj)rrds rrrms,z(() L|//0$EMMO$E ]  BJJ/0  fll ;;$ $'!* $$ $s&B8)NNr)NNN)rNN)FN).__doc__numpyrtensorflow.core.frameworkrtensorflow.python.frameworkrrrrrrr tensorflow.python.opsr r r "tensorflow.python.ops.gen_list_opsNotDifferentiablerr3r6r:r>r@rCrIrKrQrUrYRegisterGradient OperationrbrortryrrrrrrrrrrrrsL*3?.+<43+.21-../()/0 (, +( LP"&* '+%)! 2$&$( CG  8=" "*+2cmm2,2)*CS]]C+C'(QCMMQ)Q()*+ cmm ,* '(7S]]7)7,-!#--!.!*)* 3s}} 3+ 3)*-s}}-+-*()GcmmG*G () cmm * )*+, s}} -+ 9: s}} ; +%r