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 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.

    New in version 1.12.0.

Returns

y – A rotated view of m.

Return type

ndarray