It wouldn’t be possible to execute a batch of image tensors in different shapes in a single execution.
The nested tensors utility could solve it, but I’m unsure what the current status of it is.
For now, you could either pad the tensors to a common shape.