tensorflowを使って256 px$\times$256 pxのpngデータを
import tensorflow as tf
image = tf.read_file("hoge.png")
image = tf.image.decode_png(image channels=3)
で取得したらshapeが(?, ?, 3)になって困ったので結局numpyを使って
import numpy as np
img = np.array(Image.open("hoge.png"))
をしたらshapeが(256, 256, 3)で正常に得られました。
めでたしめでたし?