TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

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.