description: TensorFlow variant of NumPy's squeeze.

tf.experimental.numpy.squeeze

View source on GitHub

TensorFlow variant of NumPy's squeeze.

tf.experimental.numpy.squeeze(
    a, axis=None
)

See the NumPy documentation for numpy.squeeze.