jax.experimental.pallas.broadcast_to

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

参数：

• a (Array)

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

返回类型：

Array