tfa.image.utils.get_ndims

View source on GitHub
tfa.image.utils.get_ndims(image)