AVh!dZddlmZddlmZddlmZddlmZddlmZddl m Z e dejd Z d Z d Zy ) zSupport for ragged tensors.)dtypes) tensor_shape) ragged_config) ragged_tensor)dispatch) tf_exportzragged.map_flat_valuesc g}g}t|||}t|||}|s||i|S|r8t|}t|dk7rtd|z|j }nd}td|D}t|dkDr[t j s td|D cgc]/}|D cgc]!} | jtj#c} 1}}} ||i|} |;| jddj|gstd| jd|dtjj| t|d Scc} wcc} }w) a Applies `op` to the `flat_values` of one or more RaggedTensors. Replaces any `RaggedTensor` in `args` or `kwargs` with its `flat_values` tensor (which collapses all ragged dimensions), and then calls `op`. Returns a `RaggedTensor` that is constructed from the input `RaggedTensor`s' `nested_row_splits` and the value returned by the `op`. If the input arguments contain multiple `RaggedTensor`s, then they must have identical `nested_row_splits`. This operation is generally used to apply elementwise operations to each value in a `RaggedTensor`. Warning: `tf.ragged.map_flat_values` does *not* apply `op` to each row of a ragged tensor. This difference is important for non-elementwise operations, such as `tf.reduce_sum`. If you wish to apply a non-elementwise operation to each row of a ragged tensor, use `tf.map_fn` instead. (You may need to specify an `output_signature` when using `tf.map_fn` with ragged tensors.) Examples: >>> rt = tf.ragged.constant([[1, 2, 3], [], [4, 5], [6]]) >>> tf.ragged.map_flat_values(tf.ones_like, rt) >>> tf.ragged.map_flat_values(tf.multiply, rt, rt) >>> tf.ragged.map_flat_values(tf.add, rt, 5) Example with a non-elementwise operation (note that `map_flat_values` and `map_fn` return different results): >>> rt = tf.ragged.constant([[1.0, 3.0], [], [3.0, 6.0, 3.0]]) >>> def normalized(x): ... return x / tf.reduce_sum(x) >>> tf.ragged.map_flat_values(normalized, rt) >>> tf.map_fn(normalized, rt) Args: op: The operation that should be applied to the RaggedTensor `flat_values`. `op` is typically an element-wise operation (such as math_ops.add), but any operation that preserves the size of the outermost dimension can be used. I.e., `shape[0]` of the value returned by `op` must match `shape[0]` of the `RaggedTensor`s' `flat_values` tensors. *args: Arguments for `op`. **kwargs: Keyword arguments for `op`. Returns: A `RaggedTensor` whose `ragged_rank` matches the `ragged_rank` of all input `RaggedTensor`s. Raises: ValueError: If args contains no `RaggedTensors`, or if the `nested_splits` of the input `RaggedTensor`s are not identical. z\Input RaggedTensors' flat_values must all have the same outer-dimension size. Got sizes: %sNc3:K|]}|djyw)rN)dtype).0ps b/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/ragged/ragged_functional_ops.py z"map_flat_values..ms=1=szInput RaggedTensors have mismatched row partition dtypes; use RaggedTensor.with_row_splits_dtype() to convert them to compatible dtypes.ztf.ragged.map_flat_values requires that the output of `op` have the same outer-dimension size as flat_values of any ragged inputs. (output shape: z!; expected outer dimension size: )F)validate) _replace_ragged_with_flat_valuessetlen ValueErrorpoprauto_cast_partition_dtype with_dtyperint64shapeis_compatible_withr RaggedTensor_from_nested_row_partitions_merge_partition_lists) opargskwargspartition_listsflat_values_nrows inner_args inner_kwargspartition_dtypespartition_listr op_outputs rmap_flat_valuesr*sz//o0AC*1&/2CE,  t v  -. " C() ***--/=_== Q  2 2 4 < ==. ! "  fll # "O*- -)" ??2A  1 13D2E F  ??- / 00  # # ? ?_- @ ! "s) E2&E EEcJtj|rtj|}j|jt j |jjdj}|j||jSfdt|tr|Dcgc] }| c}St|trtfd|DSt|tr"tfd|jDS|Scc}w)aReplace RaggedTensors with their flat_values, and record their partitions. Returns a copy of `value`, with any nested `RaggedTensor`s replaced by their `flat_values` tensor. Looks inside lists, tuples, and dicts. Appends each `RaggedTensor`'s `RowPartition`s to `partition_lists`. Args: value: The value that should be transformed by replacing `RaggedTensors`. partition_lists: An output parameter used to record the row partitions for any `RaggedTensors` that were replaced. flat_values_nrows: An output parameter used to record the outer dimension size for each replacement `flat_values` (when known). Contains a list of int. Returns: A copy of `value` with nested `RaggedTensors` replaced by their `values`. rct|SN)r)vr$r#s rrecursez1_replace_ragged_with_flat_values..recurses +A,= ??c3.K|] }|ywr-)r r.r/s rrz3_replace_ragged_with_flat_values..s++sc38K|]\}}||fywr-r2)r kr.r/s rrz3_replace_ragged_with_flat_values..srNsR".466+6 #$ m%m`(V r0