@@ -0,0 +1,160 @@
+import keras.backend as K%0A%0Adef get_input(rows, cols, channels):%0A    return (channels, rows, cols) if K.image_dim_ordering() == 'th' else (rows, cols, channels)%0A

Add keras util with get input func