description: TensorFlow variant of NumPy's expand_dims.

tf.experimental.numpy.expand_dims

View source on GitHub

TensorFlow variant of NumPy's expand_dims.

tf.experimental.numpy.expand_dims(
    a, axis
)

See the NumPy documentation for numpy.expand_dims.