AVh} dZddlZddlmZddlmZddlmZddlm Z ddl m Z ddlmZdd l mZdd l mZdd l mZdd lmZdd lmZddlmZddlmZm Z m!Z!ddl"m#Z#edZ$edddZ%edddZ&dde#e!e$fde#e!e%fde#e!e&fde#e!e$ffdZ'edejPe'Z)de#e!e$fde#e!e%fde#e!e&fde#e!e$ffdZ*y)zUPython wrappers around TensorFlow ops. This file is MACHINE GENERATED! Do not edit. N) pywrap_tfe)context)core)execute)dtypes)annotation_types)op_def_registry)ops)op_def_library)deprecated_endpoints)dispatch) tf_export)TypeVarListAny) Annotated) TV_Roll_Tz_atypes.BFloat16z _atypes.Boolz_atypes.Complex128z_atypes.Complex64z_atypes.Float16z_atypes.Float32z_atypes.Float64z_atypes.Float8e4m3b11fnuzz_atypes.Float8e4m3fnz_atypes.Float8e4m3fnuzz_atypes.Float8e5m2z_atypes.Float8e5m2fnuzz _atypes.Halfz _atypes.Int16 _atypes.Int32z _atypes.Int4 _atypes.Int64z _atypes.Int8z_atypes.QInt16z_atypes.QInt32z _atypes.QInt8z_atypes.QUInt16z_atypes.QUInt8z_atypes.Resourcez_atypes.Stringz_atypes.UInt16z_atypes.UInt32z _atypes.UInt4z_atypes.UInt64z _atypes.UInt8z_atypes.VariantTV_Roll_Tshiftrr TV_Roll_Taxisinputshiftaxisreturnctjxstj}|j}|jr t j |d||||}|Stjd||||\}}} } | dd}tj rYd| j#dd| j#dd| j#df} | j$} tj&d| | ||\}|S#t j$r }tj||Yd}~nd}~wt j$rYnwxYw t|||||S#t j$rYwxYw)axRolls the elements of a tensor along an axis. The elements are shifted positively (towards larger indices) by the offset of `shift` along the dimension of `axis`. Negative `shift` values will shift elements in the opposite direction. Elements that roll passed the last position will wrap around to the first and vice versa. Multiple shifts along multiple axes may be specified. For example: ``` # 't' is [0, 1, 2, 3, 4] roll(t, shift=2, axis=0) ==> [3, 4, 0, 1, 2] # shifting along multiple dimensions # 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]] roll(t, shift=[1, -2], axis=[0, 1]) ==> [[7, 8, 9, 5, 6], [2, 3, 4, 0, 1]] # shifting along the same axis multiple times # 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]] roll(t, shift=[2, -3], axis=[1, 1]) ==> [[1, 2, 3, 4, 0], [6, 7, 8, 9, 5]] ``` Args: input: A `Tensor`. shift: A `Tensor`. Must be one of the following types: `int32`, `int64`. Dimension must be 0-D or 1-D. `shift[i]` specifies the number of places by which elements are shifted positively (towards larger indices) along the dimension specified by `axis[i]`. Negative shifts will roll the elements in the opposite direction. axis: A `Tensor`. Must be one of the following types: `int32`, `int64`. Dimension must be 0-D or 1-D. `axis[i]` specifies the dimension that the shift `shift[i]` should occur. If the same axis is referenced more than once, the total shift for that axis will be the sum of all the shifts that belong to that axis. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. RollN)namectx)rrrrTTshiftTaxis)_contextr_thread_local_datais_eagerrTFE_Py_FastPathExecute_core_NotOkStatusException_opsraise_from_not_ok_status_FallbackExceptionroll_eager_fallback_SymbolicException_op_def_library_apply_op_helper_executemust_record_gradient_get_attr_typeinputsrecord_gradient) rrrr_ctxtld_resulte__op_outputs_attrs _inputs_flats S/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/gen_manip_ops.pyrollr?siR    0h..0$ #\\ 11 fdE5$0g n(88e5t$@!QX QK' ""$3%%c*H  *G  )+F::L  fg/ (' .+  & &- ##At,,  # #    4T33  # #   s0C33D:D!!D:9D:>EE%$E%z raw_ops.Rollctj|g|g\}\}tj|g|tjtjg\}\}tj|g|tjtjg\}\}|||g}d|d|d|f} tj dd|| ||} tj rtjd|| | | \} | S)Nr r!r"sRoll)r3attrsrrr)r0args_to_matching_eager_dtypesint32int64rr1r4) rrrrr_attr_T _attr_Tshift _attr_Taxisr=r<r7s r>r,r,gs55ugsBG'8E#::E7C'--Y`YfYfIij,!88$w}}V]VcVcFfg+w%, (L'; G&   Wa F!$4 1' ""$  fg/ (' .)N)+__doc__ collectionstensorflow.pythonrtensorflow.python.eagerrr#rr'rr0tensorflow.python.frameworkrrDtensorflow.security.fuzzing.pyr_atypesr _op_def_registryr r)r r."tensorflow.python.util.deprecationr tensorflow.python.utilr _dispatch tensorflow.python.util.tf_exportrtypingrrrtyping_extensionsrrrrr? to_raw_oprr,rJr>r[sY 67179FK3IC86%%'  v )?OL/J E #y.)E)C