jax.lax.broadcast

jax.lax.broadcast(operand, sizes)

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

参数:

• operand (ArrayLike) – 数组

• sizes (Sequence[int]) – 整数序列，给出要添加到数组开头的新的前导维度的尺寸。

返回值:

包含结果的数组。

返回类型:

Array

另请参阅

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