2016-08-07 4 views

Antwort

0

Das TensorFlow OpDef Typ-System derzeit nicht unterstützt Argumente (oder Ausgänge), deren Typen entweder eine Liste von Typen oder eine einzelne Art sein kann. Stattdessen können Sie .Attr("Tout: list(type)") verwenden, um den Eingabe- (oder Ausgabetyp) darzustellen. Wenn die Unterscheidung zwischen einer Liste von Tensoren und einem einzelnen Tensor wichtig ist, können Sie einen zusätzlichen bool -bewerteten attr für diese Unterscheidung verwenden.

Beachten Sie Folgendes: Wenn Sie die automatisch generierten Python-Wrapper zum Aufrufen Ihres op verwenden und diese Eingabe einen einzelnen Tensor oder eine Liste von Tensoren akzeptieren soll, müssen Sie eine benutzerdefinierte Wrapperfunktion hinzufügen, die eine Liste erstellt wie angemessen. Dieses function in der Standard-TensorFlow-Bibliothek tut dies für QueueBase.enqueue() und QueueBase.enqueue_many() Argumente, die ein einzelner Tensor oder eine Liste von Tensoren sein können.