tf.raw_ops.Select

Выбирает элементы из x или y в зависимости от condition.

tf.raw_ops.Select(
    condition, x, y, name=None
)

Тензоры x и y должны иметь одинаковую форму; выходной тензор также будет иметь эту форму.

Тензор condition должен быть скаляром, если x и y являются скалярами. Если x и y являются векторами или имеют более высокий ранг, то condition должен быть скаляром, вектором, размер которого соответствует первому измерению x, либо иметь ту же форму, что и x.

Тензор condition действует как маска, которая на основе значения каждого элемента выбирает, следует ли взять соответствующий элемент или строку выходного тензора из x (если значение равно true) или из y (если значение равно false).

Если condition является вектором, а x и y — матрицами более высокого ранга, он выбирает строку (внешнее измерение) для копирования из x и y. Если condition имеет ту же форму, что и x и y, он выбирает элемент для копирования из x и y.

Например:

# 'condition' tensor is [[True,  False]
#                        [False, True]]
# 't' is [[1, 2],
#         [3, 4]]
# 'e' is [[5, 6],
#         [7, 8]]
select(condition, t, e)  # => [[1, 6], [7, 4]]


# 'condition' tensor is [True, False]
# 't' is [[1, 2],
#         [3, 4]]
# 'e' is [[5, 6],
#         [7, 8]]
select(condition, t, e) ==> [[1, 2],
                             [7, 8]]