unsqueeze_batch_dims

unsqueeze_batch_dims#

Signature#

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

Description#

Unsqueeze array by adding batch dimensions of size 1.

Parameters#

  • arg (Input array): axes: List of positions where to insert batch dimensions of size 1. If None, returns array unchanged.

Returns#

  • Array with batch dimensions of size 1 added at specified positions:

Contents