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 .
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2020-10-01 UTC.
[null,null,["Last updated 2020-10-01 UTC."],[],[]]