AVh*, (dZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd l mZejej  dd e j"d ej$e j"dej$e j"fdZejej& dd e j"d ej$e j"dej$e j"fdZdZdZdZdZy)z"where operation for RaggedTensors.N)ops) array_ops)math_ops)ragged_concat_ops)ragged_functional_ops)ragged_gather_ops) ragged_tensor)ragged_tensor_shape)dispatch conditionxyc|du|duk7r tdtjd||||g5tj|d}|t |cdddStj|d}tj|d}tj |||\}}}t|||cdddS#1swYyxYw)aReturn the elements where `condition` is `True`. : If both `x` and `y` are None: Retrieve indices of true elements. Returns the coordinates of true elements of `condition`. The coordinates are returned in a 2-D tensor with shape `[num_true_values, dim_size(condition)]`, where `result[i]` is the coordinates of the `i`th true value (in row-major order). : If both `x` and `y` are non-`None`: Multiplex between `x` and `y`. Choose an output shape from the shapes of `condition`, `x`, and `y` that all three shapes are broadcastable to; and then use the broadcasted `condition` tensor as a mask that chooses whether the corredsponding element in the output should be taken from `x` (if `condition` is true) or `y` (if `condition` is false). >>> # Example: retrieve indices of true elements >>> tf.where(tf.ragged.constant([[True, False], [True]])) >>> # Example: multiplex between `x` and `y` >>> tf.where(tf.ragged.constant([[True, False], [True, False, True]]), ... tf.ragged.constant([['A', 'B'], ['C', 'D', 'E']]), ... tf.ragged.constant([['a', 'b'], ['c', 'd', 'e']])) Args: condition: A potentially ragged tensor of type `bool` x: A potentially ragged tensor (optional). y: A potentially ragged tensor (optional). Must be specified if `x` is specified. Must have the same rank and type as `x`. name: A name of the operation (optional). Returns: : If both `x` and `y` are `None`: A `Tensor` with shape `(num_true, rank(condition))`. : Otherwise: A potentially ragged tensor with the same type as `x` and `y`, and whose shape is broadcast-compatible with `x`, `y`, and `condition`. Raises: ValueError: When exactly one of `x` or `y` is non-`None`; or when `condition`, `x`, and `y` have incompatible shapes. N1x and y must be either both None or both non-None RaggedWherer namer r) ValueErrorr name_scoper "convert_to_tensor_or_ragged_tensor_coordinate_wherematch_row_splits_dtypes_elementwise_where_v2r r rrs \/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/ragged/ragged_where_op.pywhere_v2rsd4iQ$Y H II ~~mTIq!+<= 4@@ %Iy y ) 4 4  : :13 Ga  : :13 Ga%==iANoiA "9a 3 4 4 4$B<AB<<Cc|du|duk7r tdtjd||||g5tj|d}|t |cdddStj|d}tj|d}tj |||\}}}t|||cdddS#1swYyxYw)a Return the elements, either from `x` or `y`, depending on the `condition`. : If both `x` and `y` are `None`: Returns the coordinates of true elements of `condition`. The coordinates are returned in a 2-D tensor with shape `[num_true_values, dim_size(condition)]`, where `result[i]` is the coordinates of the `i`th true value (in row-major order). : If both `x` and `y` are non-`None`: Returns a tensor formed by selecting values from `x` where condition is true, and from `y` when condition is false. In particular: : If `condition`, `x`, and `y` all have the same shape: * `result[i1...iN] = x[i1...iN]` if `condition[i1...iN]` is true. * `result[i1...iN] = y[i1...iN]` if `condition[i1...iN]` is false. : Otherwise: * `condition` must be a vector. * `x` and `y` must have the same number of dimensions. * The outermost dimensions of `condition`, `x`, and `y` must all have the same size. * `result[i] = x[i]` if `condition[i]` is true. * `result[i] = y[i]` if `condition[i]` is false. Args: condition: A potentially ragged tensor of type `bool` x: A potentially ragged tensor (optional). y: A potentially ragged tensor (optional). Must be specified if `x` is specified. Must have the same rank and type as `x`. name: A name of the operation (optional) Returns: : If both `x` and `y` are `None`: A `Tensor` with shape `(num_true, dim_size(condition))`. : Otherwise: A potentially ragged tensor with the same type, rank, and outermost dimension size as `x` and `y`. `result.ragged_rank = max(x.ragged_rank, y.ragged_rank)`. Raises: ValueError: When exactly one of `x` or `y` is non-`None`; or when `condition`, `x`, and `y` have incompatible shapes. #### Examples: >>> # Coordinates where condition is true. >>> condition = tf.ragged.constant([[True, False, True], [False, True]]) >>> print(where(condition)) tf.Tensor( [[0 0] [0 2] [1 1]], shape=(3, 2), dtype=int64) >>> # Elementwise selection between x and y, based on condition. >>> condition = tf.ragged.constant([[True, False, True], [False, True]]) >>> x = tf.ragged.constant([['A', 'B', 'C'], ['D', 'E']]) >>> y = tf.ragged.constant([['a', 'b', 'c'], ['d', 'e']]) >>> print(where(condition, x, y)) >>> # Row selection between x and y, based on condition. >>> condition = [True, False] >>> x = tf.ragged.constant([['A', 'B', 'C'], ['D', 'E']]) >>> y = tf.ragged.constant([['a', 'b', 'c'], ['d', 'e']]) >>> print(where(condition, x, y)) Nrrr rr r)rrrr rrr_elementwise_wherers rwherer _sN4iQ$Y H II ~~mTIq!+<= 1@@ %Iy y ) 1 1  : :13 Ga  : :13 Ga%==iANoiA  1a 0 1 1 1rct|tj}t|tj}t|tj}|s|s|stj|||S|r*|r(|r&t j tj|||S|s|jjdtj||gd}t||jj}t||jj}tj|tj||tj|z} t!j"|| St%d)z,Ragged version of tf.where(condition, x, y).raxisout_typezInput shapes do not match.) isinstancer RaggedTensorrr rmap_flat_valuesshapeassert_has_rankrconcat_nrows row_splitsdtyperrangergatherr) r r rcondition_is_ragged x_is_ragged y_is_raggedx_and_yx_nrowsy_nrowsindicess rrrs#"9m.H.HI1m889+1m889+  ??9a ++{{ 0 0)Q12 44  OO##A&&&1vA6GQ!3!3!9!9:GQ!3!3!9!9:Gooi)@%w(??AG  # #GW 55 1 22c|jjrf|jjrL|jjr2|j|jk(r|j|jk(stjj |}tjj |}tjj |}tj |tj ||}tj ||}tj ||}tj ||}t|tj}t|tj}t|tj} |s|s| stj|||Stjtj|||S)z/Ragged version of tf.where_v2(condition, x, y).)r*is_fully_definedr RaggedTensorDynamicShape from_tensorbroadcast_dynamic_shape broadcast_tor'r r(rrrr)) r r rshape_cshape_xshape_yr*r2r3r4s rrrsf // * * ,1I1I1K '' " " $AGG); //QWW $!::FFG!::FFqIG!::FFqIG  7 7$<>8> ,, ??8h 7 ::r9)NNN)__doc__typingtensorflow.python.frameworkrtensorflow.python.opsrrtensorflow.python.ops.raggedrrrr r tensorflow.python.utilr dispatch_for_apir RaggedOrDenseOptionalr rrrr-r9rrZs) ++*:>:6<+9--.?C?C=4 33=4 ; ;<=4 ; ;<=4/=4@9??+<@<@Q1]00Q1 __]88 9Q1 __]88 9Q1,Q1h3424".;r9