在張量流中創建操作時無法將屬性設置爲{類型，列表（類型）}

Question

似乎Attr("Tout: {list(type), type}")在張量流操作創建中不允許。

F tensorflow/core/framework/op.cc:160] Check failed: ::tensorflow::Status::OK() == (RegisterAlreadyLocked(deferred_[i])) (OK vs. Invalid argument: Unrecognized type string 'type' from Attr("Tout: {type, list(type)}") for Op Py 

Answer 1

的TensorFlow OpDef類型系統當前不支持參數（或輸出），其類型可以是類型列表或單一類型。相反，您可以使用.Attr("Tout: list(type)")來表示輸入（或輸出）類型。如果張量列表和單張量之間的區別很重要，則可以使用額外的attr值來表示該區別。

請注意，如果您使用自動生成的Python包裝來調用您的操作，並且您希望此輸入接受單個張量或張量列表，則需要添加一個自定義包裝函數以創建列表作爲適當的。標準TensorFlow庫中的這個function爲QueueBase.enqueue()和QueueBase.enqueue_many()參數執行此操作，該參數可以是單張量或張量列表。