tf.compat.v1.where

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

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

Этот API совместим с выполнением eager и tf.function. Однако это по-прежнему API устаревшего типа, изначально разработанный для TF1. Для миграции на полностью родной TF2 замените его использование на tf.where, который напрямую совместим со tf.compat.v1.where.

Однако tf.compat.v1.where более ограничен, чем tf.where, требуя, чтобы x и y имели одинаковую форму и возвращал тензор Tensor с тем же типом и формой, что и x, y (если они оба не None).

tf.where может принимать x, y, которые не имеют одинаковой формы, при условии, что они совместимы по трансляции друг с другом и с condition, и вернёт тензор Tensor с формой, транслируемой из condition, x, и y.

Например, следующее работает с tf.where, но не с tf.compat.v1.where:

tf.where([True, False, False, True], [1,2,3,4], [100])
<tf.Tensor: shape=(4,), dtype=int32, numpy=array([  1, 100, 100,   4],
dtype=int32)>

tf.where(True, [1,2,3,4], 100)
<tf.Tensor: shape=(4,), dtype=int32, numpy=array([1, 2, 3, 4],
dtype=int32)>

Описание

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

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

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

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