jax.lax.broadcast_to_rank

jax.lax.broadcast_to_rank(x, rank)

添加前导维度 1 使 x 的秩为 rank。

参数:

• x (ArrayLike)

• rank (int)

返回类型:

Array