squeeze_batch_dims

squeeze_batch_dims#

Signature#

nabla.squeeze_batch_dims(arg: 'Array', axes: 'list[int] | None') -> 'Array'

Description#

Squeeze array by removing batch dimensions of size 1.

Parameters#

  • arg (Input array): axes: List of batch dimension axes to squeeze. If None, returns array unchanged.

Returns#

  • Array with specified batch dimensions of size 1 removed:

Contents