Tensorflow完全に定義された形状と完全に定義された形状とで定義された多型Op

Question

TensorflowでカスタムOpを定義する場合、入力の形状が完全に定義されているかどうか多形である2つのカーネルを提供できますか？たとえば、形状が完全にわかっている/定義されている場合、カーネル構築で一度構造を構築できます。

Answer 1

これは現在できません。 kernel dispatch mechanismは、テンソル形状に関する情報が（一般的に）利用できないTensorFlowコードの低レベル部分に実装されています。

しかし、既知の図形に基づいてグラフを特殊化する能力は有用な能力のように見えますが、GitHub issues pageの機能要求としてこれを引き上げる価値があります。考えられる回避策の1つは、形状情報を利用して、既知の入力シェイプのopsの名前を静的シェイプ情報に依存する別のopに（例えば、追加のattrを使用して）書き換えることです。registering an optimization passを試すことが考えられます。ただし、TensorFlowでこれを行うには、現在、ソースから再構築する必要があります。