On this page
tf.raw_ops.Case
An n-way switch statement which calls a single branch function.
tf.raw_ops.Case(
    branch_index, input, Tout, branches, output_shapes=[], name=None
)
  An n-way switch statement, implementing the following:
  switch (branch_index) {
  case 0:
    output = branches[0](input);
    break;
  case 1:
    output = branches[1](input);
    break;
  ...
  case [[nbranches-1]]:
  default:
    output = branches[nbranches-1](input);
    break;
}
  | Args | |
|---|---|
branch_index | 
      A Tensor of type int32. The branch selector, an int32 Tensor. | 
     
input | 
      A list of Tensor objects. A list of input tensors passed to the branch function. | 
     
Tout | 
      A list of tf.DTypes. A list of output types. | 
     
branches | 
      A list of functions decorated with @Defun that has length >= 1. A list of functions each of which takes 'inputs' and returns a list of tensors, whose types are the same as what every other branch returns. | 
     
output_shapes | 
      An optional list of shapes (each a tf.TensorShape or list of ints). Defaults to []. | 
     
name | 
      A name for the operation (optional). | 
| Returns | |
|---|---|
A list of Tensor objects of type Tout. | 
     
© 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.3/api_docs/python/tf/raw_ops/Case