tfa.image.utils.to_4D_image

View source on GitHub

Convert 2/3/4D image to 4D image.

image 2/3/4D tensor.

4D tensor with the same type.