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

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.

Returns

y – A rotated view of m.

Return type

ndarray