tfa.image.utils.to_4D_image

View source on GitHub

Convert 2/3/4D image to 4D image.

tfa.image.utils.to_4D_image(image)

Args:

  • image: 2/3/4D tensor.

Returns:

4D tensor with the same type.