2016-08-17 21 views
3

私は簡単な質問tf.py_funcを持っています。Tensorflow:Py_funcは未知の形状を返します

私はイメージテンソルmy_imgの形状(1,224,224,3)を持っています。 py_funcをテストするために、テンソルを同じテンソルを返すべきPython関数return_tfに与えます(ドキュメントごとにnumpy配列に変換された後)。ここで

はコードです:

def return_tf(x): 
    return np.array(x) 

test = tf.py_func(return_tf,[my_img],[tf.float32]) 

しかし、私はtestと呼ば返さテンソルの形状を確認したときに、私が取得:

tf.Tensor 'PyFunc:0' shape=unknown dtype=float32 

私は以来、また、テンソルでeval()を実行することができませんエラーが表示されます。

AttributeError: 'list' object has no attribute 'eval'. 

テンソルのnsor形状はtf.py_funcによって返されましたか?

答えて

4

ちょうどpy_funcがtensorflowリストを返すために、私はFFを行うことができます。..回避策が見つかりました:

test = tf.reshape(tf.concat(1, test), [ <<a shape>> ]) 

所望の形状

とテンソルを取得します
関連する問題