тензорный поток :: ops :: Где3

#include <math_ops.h>

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

Резюме

Все тензоры 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]]

  

Arguments:

  • scope: A Scope object
  • x: = A Tensor which may have the same shape as condition. If condition is rank 1, x may have higher rank, but its first dimension must match the size of condition.
  • y: = A Tensor with the same type and shape as x.

Returns:

  • Output: = A Tensor with the same type and shape as x and y.

Constructors and Destructors

Where3(const ::tensorflow::Scope & scope, ::tensorflow::Input condition, ::tensorflow::Input x, ::tensorflow::Input y)

Public attributes

operation
output

Public functions

node() const
::tensorflow::Node *
operator::tensorflow::Input() const
operator::tensorflow::Output() const

Public attributes

operation

Operation operation

выход

::tensorflow::Output output

Публичные функции

Где3

 Where3(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input condition,
  ::tensorflow::Input x,
  ::tensorflow::Input y
)

узел

::tensorflow::Node * node() const 

оператор :: тензорный поток :: Вход

 operator::tensorflow::Input() const 

оператор :: тензор потока :: Вывод

 operator::tensorflow::Output() const