chainer.functions.swapaxes¶
-
chainer.functions.
swapaxes
(x, axis1, axis2)[source]¶ Swap two axes of a variable.
- Parameters
x (
Variable
or N-dimensional array) – Input variable. A \((s_1, s_2, ..., s_N)\) -shaped float array.axis1 (int) – The first axis to swap.
axis2 (int) – The second axis to swap.
- Returns
Variable whose axes are swapped.
- Return type
Example
>>> x = np.array([[[0, 1, 2], [3, 4, 5]]], np.float32) >>> x.shape (1, 2, 3) >>> y = F.swapaxes(x, axis1=0, axis2=1) >>> y.shape (2, 1, 3) >>> y.array array([[[0., 1., 2.]], <BLANKLINE> [[3., 4., 5.]]], dtype=float32)