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.