tf.raw_ops.Case
Управляющая конструкция switch, которая вызывает одну из ветвей.
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;
}
| Аргументы | |
|---|---|
branch_index | Tensor типа int32. Селектор ветви, тензор типа int32. |
input | Список Tensor объектов. Список входных тензоров, передаваемых функции ветви. |
Tout | Список tf.DTypes. Список типов вывода. |
branches | Список функций, декорированных @Defun, длиной >= 1. Список функций, каждая из которых принимает 'inputs' и возвращает список тензоров, типы которых совпадают с типами, возвращаемыми каждой из других ветвей. |
output_shapes | Необязательный список форм (каждая форма — tf.TensorShape или список ints). По умолчанию []. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Список Tensor объектов типа 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