pad_hw_dim = tf.concat([tf.zeros([1], dtype=tf.int32),
                          pad_shape - image_shape], axis=0)