def _to_bc_h_w(self, x, x_shape):
        """(b, h, w, c) -> (b*c, h, w)"""
        x = tf.transpose(x, [0, 3, 1, 2])
        x = tf.reshape(x, (-1, x_shape[1], x_shape[2]))
        return x
def _to_bc_h_w(self, x, x_shape):
        """(b, h, w, c) -> (b*c, h, w)"""
        x = tf.transpose(x, [0, 3, 1, 2])
        x = tf.reshape(x, (-1, x_shape[1], x_shape[2]))
        return x
def _to_bc_h_w ( self , x , x_shape ) : x = tf . transpose ( x , [ 0 , 3 , 1 , 2 ] ) x = tf . reshape ( x , ( - 1 , x_shape [ 1 ] , x_shape [ 2 ] ) ) return x
(b, h, w, c) -> (b*c, h, w)
( b h w c ) - > ( b * c h w )