jax.Array.squeeze#

abstractArray.squeeze(axis=None)[source]#

Remove one or more length-1 axes from array.

Refer tojax.numpy.squeeze() for full documentation.

Parameters:

• self (Array)

• axis (reductions.Axis)

Return type:

Array