tf.raw_ops.Switch
Forwards data
to the output port determined by pred
.
tf.raw_ops.Switch( data, pred, name=None )
If pred
is true, the data
input is forwarded to output_true
. Otherwise, the data goes to output_false
.
See also RefSwitch
and Merge
.
Args | |
---|---|
data | A Tensor . The tensor to be forwarded to the appropriate output. |
pred | A Tensor of type bool . A scalar that specifies which output port will receive data. |
name | A name for the operation (optional). |
Returns | |
---|---|
A tuple of Tensor objects (output_false, output_true). | |
output_false | A Tensor . Has the same type as data . |
output_true | A Tensor . Has the same type as data . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/Switch