Wie kann ich einen Teil des Graphen basierend auf einer Bedingung ausführen?
Ich habe einen Teil meines Netzwerks, der nur ausgeführt werden soll, wenn ein Platzhalterwert in feed_dict bereitgestellt wird. Ein alternativer Pfad wird verwendet, wenn der Wert nicht angegeben wird. Wie gehe ich darüber, dies mit Tensorflow zu implementieren?
Hier sind die relevanten Teile meines Codes:
%Vor% Im obigen Code suche ich nach etwas, das anstelle von if N > 0:
Hrm. Es ist möglich, dass das, was Sie wollen, ist tf.control_flow_ops.cond () Ссылка
Aber das wird nicht in den tf-Namespace exportiert, und ich antworte, ohne zu prüfen, wie garantiert-stabil diese Schnittstelle ist, aber sie wird in freigegebenen Modellen verwendet. :)
Allerdings: Da Sie im Voraus wissen, welchen Pfad Sie beim Erstellen des feed_dict haben möchten, können Sie auch einen anderen Pfad verwenden, um einen separaten Pfad durch Ihr Modell aufzurufen. Der übliche Weg, dies zu tun, besteht beispielsweise darin, Code wie folgt einzurichten:
%Vor%Und dann ziehen Sie die Knoten out1 oder out2, je nachdem, was Sie wissen, wenn Sie Ihre Berechnung ausführen und das feed_dict setzen. Denken Sie daran, dass, wenn das Modell die gleichen Variablen referenziert (erstellen Sie außerhalb die func model ()), dann haben Sie grundsätzlich zwei getrennte Pfade durch.
Ein Beispiel dafür finden Sie im Faltungsbeispiel: Ссылка
Ich bin ein Fan davon, dies zu tun, ohne Kontrollflussabhängigkeiten einzuführen, wenn Sie können.
Hier ist ein einfaches Beispiel, mit dem Sie beginnen können. Es führt verschiedene Teile des Graphen basierend auf der Form des Tensors aus:
%Vor% Ändern Sie 3 zu 2, um zu sehen, wie der Tensor subtrahiert wird. Wie Sie sehen, habe ich die Knoten für add und sub und shape initiiert, dann im Diagramm nach der Form suchen und den spezifischen Teil ausführen.
Tags und Links tensorflow