is it possible to get this working with a dynamic batch size
bg shape = (None, const, const, const)
patch1 = tf.image.extract_patches(bg,
sizes=[1, self.k_size, self.k_size, 1],
strides=[1, self.stride, self.stride, 1],
rates=[1, 1, 1, 1],
padding='VALID')