tf.experimental.numpy.expand_dims

View source on GitHub

TensorFlow variant of NumPy's expand_dims.

See the NumPy documentation for numpy.expand_dims.