jax.experimental.pallas.broadcast_to

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

Parameters:

• a (Array)

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

Return type:

Array