tf.keras.backend.expand_dims

TensorFlow 2 version View source on GitHub

Adds a 1-sized dimension at index "axis".

x A tensor or variable.
axis Position where to add a new axis.

A tensor with expanded dimensions.