jax.Array.repeat

abstract Array.repeat(repeats, axis=None, *, total_repeat_length=None)

從重複的元素建構陣列。

完整文件請參閱 jax.numpy.repeat()。

參數：

• self (Array)

• repeats (ArrayLike)

• axis (int | None)

• total_repeat_length (int | None)

返回類型：

Array