jax.Array.repeat

jax.Array.repeat#

abstract Array.repeat(repeats, axis=None, *, total_repeat_length=None)[source]#

从重复的元素构建数组。

请参考 jax.numpy.repeat() 获取完整文档。

参数:
  • self (Array)

  • repeats (ArrayLike)

  • axis (int | None)

  • total_repeat_length (int | None)

返回类型:

Array