tf.where_v2
Возвращает элементы, либо из x, либо из y, в зависимости от condition.
tf.where_v2(
condition, x=None, y=None, name=None
)
Если оба x и y равны None, то эта операция возвращает координаты истинных элементов condition. Координаты возвращаются в 2-мерном тензоре, где первая размерность (строки) представляет количество истинных элементов, а вторая размерность (столбцы) представляет координаты истинных элементов. Имейте в виду, что форма выходного тензора может изменяться в зависимости от того, сколько истинных значений содержится во входных данных. Индексы выводятся в порядке следования строк.
Если оба не равны None, condition, x и y должны быть согласуемы по форме.
Тензор condition действует как маска, которая, в зависимости от значения в каждом элементе, выбирает, брать ли соответствующий элемент/строку из x (если истинно) или y (если ложно).
| Аргументы | |
|---|---|
condition | A Tensor типа bool |
x | Тензор того же типа, что и y, и может быть согласован по форме с condition и y. |
y | Тензор того же типа, что и x, и может быть согласован по форме с condition и x. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor того же типа, что и x и y, и формой, согласованной с condition, x, и y, если x, y не равны None. A Tensor с формой (num_true, dim_size(condition)). |
| Исключения | |
|---|---|
ValueError | При точном значении, что x или y не равны None. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/where_v2
