jax.lax.squeeze

jax.lax.squeeze(array, dimensions)[source]

Squeeze any number of size 1 dimensions from an array.

Parameters
Return type

Any