tensorflow :: opérations :: RefSwitch

Transmet les data tenseur de référence au port de sortie déterminé par pred .

Résumé

Si pred est vrai, l'entrée de data est transmise à output_true . Sinon, les données vont à output_false .

Voir également Switch et Merge .

Arguments:

• scope: un objet Scope
• data: Le tenseur ref à transmettre à la sortie appropriée.
• pred: un scalaire qui spécifie le port de sortie qui recevra les données.

Retour:

• Output output_false: Si pred est false, les données seront transmises à cette sortie.
• Output output_true: Si pred est vrai, les données seront transmises à cette sortie.

Attributs publics

Operation operation

::tensorflow::Output output_false

::tensorflow::Output output_true

 RefSwitch(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input data,
  ::tensorflow::Input pred
)