tfa.image.utils.from_4D_image

View source on GitHub

Convert back to an image with ndims rank.

image 4D Tensor.
ndims The original rank of the image.

ndims-D Tensor with the same type.