2016-12-24 23 views
1

私はテンソルで最大プールします。しかし、組み込みのtf.nn.max_pool()を使う代わりに、tf.reduce_max()を使う必要があります。しかし、それはエラーを与える:形状はランク4でなく、ランク5でなければなりません。

ValueError: Shape must be rank 4 but is rank 5 for 'conv2_1/Conv2D' (op: 'Conv2D') with input shapes: [1,?,1,224,64], [3,3,64,128]. 

そしてここではコードです:

with tf.name_scope('conv1_2') as scope: 
     kernel = tf.Variable(tf.truncated_normal([3, 3, 64, 64], dtype=tf.float32, 
               stddev=1e-1), name='weights') 
     conv = tf.nn.conv2d(self.conv1_1, kernel, [1, 1, 1, 1], padding='SAME') 
     biases = tf.Variable(tf.constant(0.0, shape=[64], dtype=tf.float32), 
          trainable=True, name='biases') 
     out = tf.nn.bias_add(conv, biases) 
     self.conv1_2 = tf.nn.relu(out, name=scope) 
     self.parameters += [kernel, biases] 

self.pool1=tf.reduce_max(self.conv1_2,reduction_indices=[1], keep_dims=True) 


     # conv2_1 
     with tf.name_scope('conv2_1') as scope: 
      kernel = tf.Variable(tf.truncated_normal([3, 3, 64, 128], dtype=tf.float32, 
                stddev=1e-1), name='weights') 
     sess = tf.InteractiveSession() 
     tf.Print(self.pool1,[self.pool1],message="hellow fatima") 
      conv = tf.nn.conv2d([self.pool1], kernel, [1, 1, 1, 1], padding='SAME') 
      biases = tf.Variable(tf.constant(0.0, shape=[128], dtype=tf.float32), 
           trainable=True, name='biases') 
      out = tf.nn.bias_add(conv, biases) 
      self.conv2_1 = tf.nn.relu(out, name=scope) 
      self.parameters += [kernel, biases] 

答えて

1

あなたはtf.nn.conv2dに渡される最初のパラメータに[]を削除しようとするかもしれません。したがって、conv = tf.nn.conv2d([self.pool1], kernel, [1, 1, 1, 1], padding='SAME')の代わりに、これを試してください:conv = tf.nn.conv2d(self.pool1, kernel, [1, 1, 1, 1], padding='SAME')

関連する問題