tf.keras.backend.switch
Переключает две операции в зависимости от скалярного значения.
tf.keras.backend.switch(
condition, then_expression, else_expression
)
Обратите внимание, что как then_expression , так и else_expression должны быть символьными тензорами одного и того же размера.
| Аргументы |
|---|
condition | тензор (int или bool). |
then_expression | либо тензор, либо вызываемая функция, возвращающая тензор. |
else_expression | либо тензор, либо вызываемая функция, возвращающая тензор. |
| Возвращаемое значение |
|---|
| Выбранный тензор. |
| Исключения |
|---|
ValueError | Если ранг condition больше ранга выражений. |
