image tensor shape = (561, 561, 3)
convert/transform tensor shape as (3, 1, 561, 561)
Now get three images from tensor.

NOTE: If tensor is (561, 561, 4) then converting shape should be (6, 1, 561, 561) and we should get 6 images.
Thanks in advance.