jax.experimental.pallas.broadcast_to

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

参数:

• a (Array)

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

返回类型:

Array