jax.lax.broadcast

jax.lax.broadcast(operand, sizes, *, out_sharding=None)

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

参数:

• operand (ArrayLike) – 一个数组

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

返回:

一个包含结果的数组。

返回类型:

Array (数组)

另请参阅

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