move_axis_to_front_of_batch_dims#
Signature#
nabla.move_axis_to_front_of_batch_dims(input_array: nabla.core.array.Array, axis: int) -> nabla.core.array.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)