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