Good question. I would like to know that too. The only thing that I was able to come up with is creating batches 16x1x512x512 and then cropping them to the desired size. But it’s terribly inefficient.