jax.numpy.rot90

jax.numpy.rot90(m, k=1, axes=(0, 1))

在 axes 指定的平面中將陣列逆時針旋轉 90 度。

numpy.rot90() 的 JAX 實作。

參數:

• m (ArrayLike) – 輸入陣列。必須具有 m.ndim >= 2。

• k (int) – int，選填，預設值=1。指定陣列旋轉的次數。對於 k 的負值，陣列會以順時針方向旋轉。

• axes (tuple[int, int]) – 2 個整數的元組，選填，預設值= (0, 1)。axes 定義陣列旋轉所在的平面。兩個軸必須不同。

回傳:

包含輸入 m 旋轉 90 度的副本的陣列。

回傳類型:

Array

另請參閱

• jax.numpy.flip()：反轉沿給定軸的順序

• jax.numpy.fliplr()：反轉沿軸 1 的順序（左右）

• jax.numpy.flipud()：反轉沿軸 0 的順序（上下）

範例

>>> m = jnp.array([[1, 2, 3],
...                [4, 5, 6]])
>>> jnp.rot90(m)
Array([[3, 6],
       [2, 5],
       [1, 4]], dtype=int32)
>>> jnp.rot90(m, k=2)
Array([[6, 5, 4],
       [3, 2, 1]], dtype=int32)

jnp.rot90(m, k=1, axes=(1, 0)) 等同於 jnp.rot90(m, k=-1, axes(0,1))。

>>> jnp.rot90(m, axes=(1, 0))
Array([[4, 1],
       [5, 2],
       [6, 3]], dtype=int32)
>>> jnp.rot90(m, k=-1, axes=(0, 1))
Array([[4, 1],
       [5, 2],
       [6, 3]], dtype=int32)

當輸入陣列的 ndim>2 時

>>> m1 = jnp.array([[[1, 2, 3],
...                  [4, 5, 6]],
...                 [[7, 8, 9],
...                  [10, 11, 12]]])
>>> jnp.rot90(m1, k=1, axes=(2, 1))
Array([[[ 4,  1],
        [ 5,  2],
        [ 6,  3]],

       [[10,  7],
        [11,  8],
        [12,  9]]], dtype=int32)