tf.experimental.numpy.ndim

TensorFlow variant of NumPy's ndim.