jax.lax.broadcast

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

廣播陣列，新增前導維度

參數:

• operand (ArrayLike) – 一個陣列

• sizes (Sequence[int]) – 整數序列，指定要新增到陣列前端的新前導維度的大小。

回傳:

包含結果的陣列。

回傳類型:

Array

參見

jax.lax.broadcast_in_dim：在陣列形狀的任何位置新增維度。