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