Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfa.image.utils.get_ndims

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