jax.lax.squeeze# jax.lax.squeeze(array, dimensions)[source]# Squeeze any number of size 1 dimensions from an array. Parameters: array (jax.typing.ArrayLike) dimensions (Sequence[int]) Return type: Array
