Hello My input is [batch, H, W, C]. I see that the visualization only output result for first element in the batch, like [0, H, W, C]. Is there a way to specify which element to visualize ?? Best He Zhao