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 (если истинно) или из y (если ложно).

Если 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]]