jax.numpy.broadcast_to

jax.numpy.broadcast_to#

jax.numpy.broadcast_to(array, shape)[源代码][源代码]#

将数组广播到指定形状。

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

参数:

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

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

返回:

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

返回类型:

Array

参见

示例

>>> 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)
目录