tf.where

Возвращает элементы, либо из x, либо из y, в зависимости от condition. (устаревшее)

tf.where(
    condition, x=None, y=None, name=None
)

Если и x, и y равны None, то данная операция возвращает координаты элементов с истинным значением в condition. Координаты возвращаются в двумерном тензоре, где первая размерность (строки) представляет количество истинных элементов, а вторая размерность (столбцы) представляет координаты истинных элементов. Имейте в виду, что форма выходного тензора может меняться в зависимости от количества истинных значений на входе. Индексы выводятся в порядке следования строк (row-major order).

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

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

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