jax.numpy.swapaxesΒΆ
-
jax.numpy.
swapaxes
(a, axis1, axis2)[source]ΒΆ Interchange two axes of an array.
LAX-backend implementation of
swapaxes()
. Original docstring below.- Parameters
- Returns
a_swapped β For NumPy >= 1.10.0, if a is an ndarray, then a view of a is returned; otherwise a new array is created. For earlier NumPy versions a view of a is returned only if the order of the axes is changed, otherwise the input array is returned.
- Return type
Examples
>>> x = np.array([[1,2,3]]) >>> np.swapaxes(x,0,1) array([[1], [2], [3]])
>>> x = np.array([[[0,1],[2,3]],[[4,5],[6,7]]]) >>> x array([[[0, 1], [2, 3]], [[4, 5], [6, 7]]])
>>> np.swapaxes(x,0,2) array([[[0, 4], [2, 6]], [[1, 5], [3, 7]]])