jax.lax.broadcast

jax.lax.broadcast(operand, sizes)

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

参数:

• operand (ArrayLike) – 一个数组

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

返回:

包含结果的数组。

返回类型:

Array

参见

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