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]]