tfa.image.utils.get_ndims

View source on GitHub