@private_method<EOL><INDENT>def _to_b_h_w_n_c(self, x, x_shape):<DEDENT>
x = tf.reshape(x, (-<NUM_LIT:1>, x_shape[<NUM_LIT:4>], x_shape[<NUM_LIT:1>], x_shape[<NUM_LIT:2>], x_shape[<NUM_LIT:3>]))<EOL>x = tf.transpose(x, [<NUM_LIT:0>, <NUM_LIT:2>, <NUM_LIT:3>, <NUM_LIT:4>, <NUM_LIT:1>])<EOL>return x<EOL>
(b*c, h, w, n) -> (b, h, w, n, c)