jax.Array.squeeze

abstract Array.squeeze(axis=None)

從陣列中移除一個或多個長度為 1 的軸。

請參閱 jax.numpy.squeeze() 以取得完整文件。

參數:

• self (Array)

• axis (reductions.Axis)

回傳類型:

Array