flux tensoriel : : opérations : : Changer

#include <control_flow_ops.h>

Transfère data vers le port de sortie déterminé par pred .

Résumé

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

Voir également RefSwitch et Merge .

Arguments:

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

Retour:

  • Output Output_false : Si pred est faux, les données seront transmises à cette sortie.
  • Output : Si pred est vrai, les données seront transmises à cette sortie.

Constructeurs et Destructeurs

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

Attributs publics

operation
output_false
output_true

Attributs publics

opération

Operation operation

sortie_false

::tensorflow::Output output_false

sortie_true

::tensorflow::Output output_true

Fonctions publiques

Changer

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