jax.experimental.pallas.broadcast_to

jax.experimental.pallas.broadcast_to(a, shape)

參數:

• a (Array)

• shape (tuple[int, ...])

回傳類型:

Array