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.