jax.lax.squeeze# jax.lax.squeeze(array, dimensions)[source]# Squeeze any number of size 1 dimensions from an array. Parameters: array (Union[Array, ndarray, bool_, number, bool, int, float, complex]) dimensions (Sequence[int]) Return type: Array