tf.raw_ops.Case
Stay organized with collections
Save and categorize content based on your preferences.
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.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2020-10-01 UTC."],[],[]]