jax.lax.broadcast

jax.lax.broadcast(operand, sizes, sharding=None)

广播一个数组，添加新的前导维度。

参数:

• operand (ArrayLike) – 一个数组

• sizes (Sequence[int]) – 一个整数序列，指定要添加到数组前面的新的前导维度的大小。

返回:

包含结果的数组。

返回类型:

Array

另请参阅

jax.lax.broadcast_in_dim：在数组形状的任何位置添加新的维度。