BVhpdZddlZddlmZddlmZddlmZddlmZddlmZddlm Z Gd d ejd d Z d Z d Z dZdZdZdZdZdZdZdZdZy)z)Operators specific to slicing operations.N)dtypes) tensor_util) gen_array_ops)gen_string_ops)list_ops)tensor_array_opsc eZdZy) GetItemOptsN)__name__ __module__ __qualname__\/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/autograph/operators/slices.pyr r srr  element_dtypect|tsJt|tjr t ||St j |rx|jtjk(r t|||S|jtjk(r%|jjdk(r t||St||St!||S)aThe slice read operator (i.e. __getitem__). Note: it is unspecified whether target will be mutated or not. In general, if target is mutable (like Python lists), it will be mutated. Args: target: An entity that supports getitem semantics. i: Index to read from. opts: A GetItemOpts object. Returns: The read element. Raises: ValueError: if target is not of a supported type. r) isinstancer r TensorArray_tf_tensorarray_get_itemr is_tf_typedtypervariant_tf_tensor_list_get_itemstringshapendims_tf_tensor_string_get_item_tf_tensor_get_item _py_get_item)targetioptss rget_itemr$"s" D+ && &(445 #FA ..f% ||v~~% %fa 66  &6<<+=+=+B ' 22  ++  ""rc$|j|S)z4Overload of get_item that stages a TensorArray read.)readr!r"s rrrBs Qrcx|j tdtj|||j}|S)z4Overload of get_item that stages a Tensor list read.zacannot retrieve from a list without knowing its element type; use set_element_type to annotate itr)r ValueErrorrtensor_list_get_item)r!r"r#xs rrrGsB  I JJ##FAT=O=OP! (rc ||S)zAOverload of get_item that stages a Tensor (not Tensor list) read.rr's rrrP rc4tj||d}|S)z6Overload of get_item that stages a Tensor string read.)rsubstrr!r"r+s rrrUsFAq)! (rc ||S)z>Overload of get_item that executes a Python list modification.rr's rr r [r-rct|tjr t|||St j |r7|j tjk(r t|||St|||St|||S)aThe slice write operator (i.e. __setitem__). Note: it is unspecified whether target will be mutated or not. In general, if target is mutable (like Python lists), it will be mutated. Args: target: An entity that supports setitem semantics. i: Index to modify. x: The new element value. Returns: Same as target, after the update was performed. Raises: ValueError: if target is not of a supported type. ) rrr_tf_tensorarray_set_itemrrrrr_tf_tensor_list_set_item_tf_tensor_set_item _py_set_itemr1s rset_itemr8`so"(445 #FAq 11f% ||v~~% %fa 33 A .. 1 %%rc&|j||S)z5Overload of set_item that stages a TensorArray write.)writer1s rr4r4|s a rc0tj|||S)z6Overload of set_item that stages a Tensor list update.)rtensor_list_set_itemr1s rr5r5s  & &vq! 44rc6tj||ff|fS)z9Overload of set_item that stages a Tensor scatter update.)rtensor_scatter_updater1s rr6r6s  , ,VqdWqd CCrc|||<|S)z>Overload of set_item that executes a Python list modification.rr1s rr7r7s&) -r)__doc__ collectionstensorflow.python.frameworkrrtensorflow.python.opsrrrr namedtupler r$rrrrr r8r4r5r6r7rrrrEsu0.3/0*2 (+((8JK#@     &8 5 D r