move_axis_from_front

move_axis_from_front#

Signature#

nabla.move_axis_from_front(input_array: nabla.core.array.Array, target_axis: int) -> nabla.core.array.Array

Description#

Move front axis (position 0) to specified target position.

Parameters#

input_array: Input tensor (assumes front axis is the one to move) target_axis: Target position for the front axis

Returns#

Tensor with front axis moved to target position

Examples#

x = nb.ones((4, 2, 3)) # front axis has size 4 y = move_axis_from_front(x, 2) # shape (2, 3, 4)

front axis moved to position 2: [1, 2, 0]