tf.keras.ops.where
Return elements chosen from x1 or x2 depending on condition.
tf.keras.ops.where(
condition, x1=None, x2=None
)
| Args | |
|---|---|
condition | Where True, yield x1, otherwise yield x2. |
x1 | Values from which to choose when condition is True. |
x2 | Values from which to choose when condition is False. |
| Returns | |
|---|---|
A tensor with elements from x1 where condition is True, and elements from x2 where condition is False. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/where
