tfa.image.utils.from_4D_image

View source on GitHub

Convert back to an image with ndims rank.

tfa.image.utils.from_4D_image(
    image,
    ndims
)

Args:

  • image: 4D tensor.
  • ndims: The original rank of the image.

Returns:

ndims-D tensor with the same type.