BVh-@dZddlZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z d d Z d!d Z d!d ZdZdZdZdZdZGddej(ddZdZdZdZGddej(ddZdZdZdZdZy)"zDOperators specific to data structures: list append, subscripts, etc.N) constant_op)dtypes)ops) tensor_util) array_ops)cond)list_ops)tensor_array_opscR|r t|}nd}|r t|St|S)zThe list constructor. Args: iterable: Optional elements to fill the list with. Returns: A list-like object. The exact return value depends on the initial elements. )tuple _py_list_newtf_tensor_list_new)iterableelementss e/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/autograph/operators/data_structures.pynew_listr s0XHH   !! H %%ctd|D}td|D}t|dk(r0t|\}|V||k7rQtdj |||t|dkDrtdj || tdtd|D}t|dk(r0t|\}|K||k7rFtd j |||t|dkDrtd j |d}|}||}t j |t|d |du| }t|D]\}} |j|| }|S) 8Overload of new_list that stages a Tensor list creation.c3FK|]}tj|ywNrconvert_to_tensor.0els r z&tf_tensor_array_new..7@3((,@!c34K|]}|jywrdtypers rrz&tf_tensor_array_new..9/288/N7incompatible dtype; specified: {}, inferred from {}: {}z.H"@5))+,@-/7incompatible shape; specified: {}, inferred from {}: {}z| tdtj|dd}t j ||}|Std|D}td|D}t|dk(r1t|d}|a||k7r\tdj||||r.|td j||tj}ntj}td |D}t|dk(r;tj|d}|k||k7rftd j||||r3|td j||tjd }ntjd }||}||}tj |tj"}t j$||}|D]}t j&||}|S)rNzAelement shape may not be specified when creating list from tensorr&)r2c3FK|]}tj|ywrrrs rrz%tf_tensor_list_new..qrr c34K|]}|jywrr"rs rrz%tf_tensor_list_new..sr$r%rr'z;specified dtype {} is inconsistent with that of elements {}c3bK|]'}t|jj)ywrr)rs rrz%tf_tensor_list_new..r,r-r.z;specified shape {} is inconsistent with that of elements {}r"r2r:)r is_tf_typer5rr*r tensor_list_from_tensorr r3r4r6rvariantrconstantrrint32empty_tensor_listtensor_list_push_back) rr:r2r?r;r<r=r>rs rrrgsH%  M OOOOH-ab1M((OA H @x@ @(/h//*_:&q)N ]n%D  C J JX~ 7 88  G N NX ' ((^^N^^N@x@@*___Xa[1N ]n%D  C J JX~ 7 88  G N NX ' ((!))"-N ))"-N"M"M'' V\\J-  !@! .b&&q"-A. (rct|S)z0Overload of new_list that creates a Python list.)list)rs rrrs hrct|tjr t||St j |r7|j tjk(r t||Std|zt||S)a The list append function. Note: it is unspecified where list_ will be mutated or not. If list_ is a TensorFlow entity, it will not be typically mutated. If list_ is a plain list, it will be. In general, if the list is mutated then the return value should point to the original entity. Args: list_: An entity that supports append semantics. x: The element to append. Returns: Same as list_, after the append was performed. Raises: ValueError: if list_ is not of a known list-like type. Otensor lists are expected to be Tensors with dtype=tf.variant, instead found %s) isinstancer r7_tf_tensorarray_appendrrHr#rrJ_tf_tensor_list_appendr5_py_list_appendlist_xs r list_appendrZsv$'334 !% ++e$ {{fnn$ #E1 --   % & '' 5! $$rcfd}tjtjdkDfd|tjS)z8Overload of list_append that stages a Tensor list write.ctj}tjt j ||j S)NrG)rrr rMrr*r#)tensor_xrYs rempty_list_of_elements_like_xz=_tf_tensor_list_append..empty_list_of_elements_like_xs9$$Q'H  % %ooh/nn &&rrcSrr rXsrz(_tf_tensor_list_append..ser)rr tensor_list_lengthrN)rXrYr^s`` rrUrUsF& ))!!%(1,# %  ' 'q 11rcB|j|j|S)z8Overload of list_append that stages a TensorArray write.)r9r/rWs rrTrTs UZZ\1 %%rc(|j||S)z;Overload of list_append that executes a Python list append.)appendrWs rrVrVs,,q/ ,rc eZdZy) ListPopOptsN__name__ __module__ __qualname__r rrrgrgsrrg)r:r2c"t|tsJt|tjr t dt j |r8|jtjk(r t|||St d|zt||S)aThe list pop function. Note: it is unspecified where list_ will be mutated or not. If list_ is a TensorFlow entity, it will not be typically mutated. If list_ is a plain list, it will be. In general, if the list is mutated then the return value should point to the original entity. Args: list_: An entity that supports pop semantics. i: Optional index to pop from. May be None. opts: A ListPopOpts. Returns: Tuple (x, out_list_): out_list_: same as list_, after the removal was performed. x: the removed element value. Raises: ValueError: if list_ is not of a known list-like type or the operation is not supported for that type. z)TensorArray does not support item removalrR) rSrgr r7r5rrHr#rrJ_tf_tensor_list_pop _py_list_pop)rXr@optss rlist_poprps, D+ && &'334 @ AAe$ {{fnn$ 4 00   % & '' q !!rc| td|j td|j tdt j ||j\}}|j |j||fS)z3Overload of list_pop that stages a Tensor list pop.z/tensor lists only support removing from the endz\cannot pop from a list without knowing its element type; use set_element_type to annotate itz]cannot pop from a list without knowing its element shape; use set_element_type to annotate itr:)NotImplementedErrorr:r5r2r tensor_list_pop_back set_shape)rXr@rolist_outrYs rrmrm s] O PP  A BB  B CC-- 4--/+(A++d  ! 1rcX||j}||fS|j|}||fS)z8Overload of list_pop that executes a Python list append.)pop)rXr@rYs rrnrns5Y A /  ! A /rc eZdZy) ListStackOptsNrhr rrrzrz'srrz)r: original_callct|tsJt|tjr t |St j |r+|jtjk(r t||S|St||S)aThe list stack function. This does not have a direct correspondent in Python. The closest idiom to this is tf.append or np.stack. It's different from those in the sense that it accepts a Tensor list, rather than a list of tensors. It can also accept TensorArray. When the target is anything else, the dispatcher will rely on ctx.original_call for fallback. Args: list_: An entity that supports append semantics. opts: A ListStackOpts object. Returns: The output of the stack operation, typically a Tensor. ) rSrzr r7_tf_tensorarray_stackrrHr#rrJ_tf_tensor_list_stack_py_list_stackrXros r list_stackr-so D- (( ('334  ''e$ {{fnn$ "5$ //l % &&rc"|jS)z7Overload of list_stack that stages a TensorArray stack.)stackr`s rr}r}Ks rcr|j tdtj||jS)z7Overload of list_stack that stages a Tensor list write.zYcannot stack a list without knowing its element type; use set_element_type to annotate itrr)r:r5r tensor_list_stackrs rr~r~Ps9  < ==  # #E9K9K LLrc$|j|S)z:Overload of list_stack that executes a Python list append.)r{rs rrrXs   E ""rr)NN)__doc__ collectionstensorflow.python.frameworkrrrrtensorflow.python.opsrrr r rrArrrZrUrTrV namedtuplergrprmrnrzrr}r~rr rrrsK3.+3+&*2 &*/ d8 v %> 2 & K=*LM ""J"K?=? '< M#r