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 больше ранга выражений. |
