jax.numpy.resize

jax.numpy.resize(a, new_shape)

返回一个具有指定形状的新数组。

LAX 后端实现 numpy.resize()。

原始文档字符串如下。

如果新数组比原数组大，那么新数组会用 a 的重复副本填充。注意，这种行为与 a.resize(new_shape) 不同，后者用零填充而不是 a 的重复副本。

参数:

• a (array_like) – 要调整大小的数组。

• new_shape (int or tuple of int) – 调整大小后的数组形状。

返回:

reshaped_array – 新数组由旧数组中的数据形成，必要时重复以填充所需数量的元素。数据按C顺序迭代重复。

返回类型:

ndarray