jax.lax.broadcast_to_rank

jax.lax.broadcast_to_rank(x, rank)

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

参数:

• x (类数组)

• rank (int)

返回类型:

数组