move_axis_to_front_of_batch_dims

move_axis_to_front_of_batch_dims#

Signature#

nabla.move_axis_to_front_of_batch_dims(input_array: 'Array', axis: 'int') -> 'Array'

Description#

Move specified batch dimension to the front (position 0), shifting others right.

Parameters#

  • input_array (Input tensor with batch dimensions): axis: Batch dimension to move to front (negative index)

Returns#

  • Tensor with specified batch dimension moved to front:

Examples#

>>> x = nb.ones((2, 3, 4))  # shape (2, 3, 4)
    >>> x.batch_dims = (1, 0)  # Simulated for example
    >>> y = move_axis_to_fron_of_batch_dims(x, -1)  # Move last batch dim to front
    >>> # Result has batch_dims=(0, 1) and shape=(2, 3, 4)
Contents