jax.numpy.rot90#

jax.numpy.rot90(m, k=1, axes=(0, 1))[source]#

Rotate an array by 90 degrees in the plane specified by axes.

LAX-backend implementation of numpy.rot90().

The JAX version of this function may in some cases return a copy rather than a view of the input.

Original docstring below.

Rotation direction is from the first towards the second axis.

Parameters
  • m (array_like) – Array of two or more dimensions.

  • k (integer) – Number of times the array is rotated by 90 degrees.

  • axes ((2,) array_like) – The array is rotated in the plane defined by the axes. Axes must be different.

Returns

y – A rotated view of m.

Return type

ndarray