jax.numpy.tile

jax.numpy.tile(A, reps)

透過沿指定維度重複 A 來建構陣列。

numpy.tile() 的 JAX 實作。

如果 A 是形狀為 (d1, d2, ..., dn) 的陣列，且 reps 是一個整數序列，則產生的陣列形狀將為 (reps[0] * d1, reps[1] * d2, ..., reps[n] * dn)，其中 A 沿每個維度平鋪。

參數：

• A (ArrayLike) – 要重複的輸入陣列。可以是任何形狀或維度。

• reps (DimSize | Sequence[DimSize]) – 指定沿每個軸的重複次數。

傳回：

根據 reps 重複輸入陣列的新陣列。

傳回類型：

Array

參見

• jax.numpy.repeat()：從重複元素建構陣列。

• jax.numpy.broadcast_to()：將陣列廣播到指定的形狀。

範例

>>> arr = jnp.array([1, 2])
>>> jnp.tile(arr, 2)
Array([1, 2, 1, 2], dtype=int32)
>>> arr = jnp.array([[1, 2],
...                  [3, 4,]])
>>> jnp.tile(arr, (2, 1))
Array([[1, 2],
       [3, 4],
       [1, 2],
       [3, 4]], dtype=int32)