jax.Array.repeat

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

从重复元素构造数组。

有关完整文档，请参阅 jax.numpy.repeat()。

参数:

• self (Array)

• repeats (ArrayLike)

• axis (int | None)

• total_repeat_length (int | None)

• out_sharding (NamedSharding | PartitionSpec | None)

返回类型:

Array