jax.lax.squeeze

Warning

This page was created from a pull request (#9655).

jax.lax.squeeze¶

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

Squeeze any number of size 1 dimensions from an array.

Parameters

array (Any) –
dimensions (Sequence[int]) –

Return type

previous

jax.lax.square

next

jax.lax.sub