jax.numpy.broadcast_to

jax.numpy.broadcast_to(array, shape)

将数组广播到指定的形状。

JAX 实现的 numpy.broadcast_to()。JAX 使用 NumPy 风格的广播规则，您可以在 NumPy 广播中了解更多信息。

参数：

• array (ArrayLike) – 要广播的数组。

• shape (DimSize | Shape) – 数组将要广播到的形状。

返回:

一个广播到指定形状的数组副本。

返回类型:

数组

另请参阅

• jax.numpy.broadcast_arrays(): 将数组广播到通用形状。

• jax.numpy.broadcast_shapes(): 将输入形状广播到通用形状。

示例

>>> x = jnp.int32(1)
>>> jnp.broadcast_to(x, (1, 4))
Array([[1, 1, 1, 1]], dtype=int32)

>>> x = jnp.array([1, 2, 3])
>>> jnp.broadcast_to(x, (2, 3))
Array([[1, 2, 3],
       [1, 2, 3]], dtype=int32)

>>> x = jnp.array([[2], [4]])
>>> jnp.broadcast_to(x, (2, 4))
Array([[2, 2, 2, 2],
       [4, 4, 4, 4]], dtype=int32)