|---| | [ View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.15.1/tensorflow/python/ops/numpy_ops/np_array_ops.py#L875-L879) | TensorFlow variant of NumPy's `expand_dims`. ```python tf.experimental.numpy.expand_dims( a, axis ) ``` See the NumPy documentation for [`numpy.expand_dims`](https://numpy.org/doc/stable/reference/generated/numpy.expand_dims.html).