def forward(self): temp = tf.transpose( self.inp.out, [0,3,1,2]) self.out = slim.flatten( temp, scope = self.scope)