jax.numpy.tile

jax.numpy.tile(A, reps)

通过将 A 重复 reps 指定的次数来构造一个数组。

LAX 后端实现 numpy.tile()。

原始文档字符串如下。

如果 reps 的长度为 d，则结果将具有 max(d, A.ndim) 的维度。

如果 A.ndim < d，则通过预先添加新轴将 A 提升为 d 维。因此，对于形状为 (3,) 的数组，对于 2 维复制将其提升为 (1, 3)，或者对于 3 维复制将其提升为形状 (1, 1, 3)。如果这不是所需的 behavior，请在调用此函数之前手动将 A 提升到 d 维。

如果 A.ndim > d，则通过在其前面添加 1 将 reps 提升为 A.ndim。因此，对于形状为 (2, 3, 4, 5) 的 A，形状为 (2, 2) 的 reps 将被视为 (1, 1, 2, 2)。

注意：虽然 tile 可用于广播，但强烈建议使用 numpy 的广播操作和函数。

参数：

• A (array_like) – 输入数组。

• reps (array_like) – 沿每个轴重复 A 的次数。

返回值：

c – 拼接后的输出数组。

返回类型:

ndarray