2017-12-05 6 views
0

形状が[なし、寸法]のプレースホルダーがあり、「なし」はバッチサイズを意味します。私は本当の形を "なし"にしたいと思っています。TensorFlowでNone(動的入力シェイプ)のリアルな形状を取得する方法は?

私はモデルを構築する際に二つの方法を試してみてください。

まず、x.get_shape()となどの形状を取得[寸法(なし)、寸法(128)]

第二に、x.shapeと取得

私が望むのは、このラウンドでバッチサイズが100であるなど、実際の形状です。[Dimension(100)、次元(128)]。

動的入力シェイプを取得するにはどうすればよいですか?

答えて

1

私はtf.shapeがあなたが探していると信じています。

tf.shape(x)セッションが実行されている間に形状を取得できます。

完全な例は以下の通りです:

import tensorflow as tf 
a = tf.ones([3,4]) 
b = tf.shape(a) 
sess=tf.Session() 
print(b.eval(session=sess)) 

ます。また、新しい変数を初期化するためにbを使用することができます。

+0

tf.shape(x)return(2、)、これは次元数を意味します。 – yck

+0

@ysk checkout私の編集済みの全回答 –

関連する問題