jax.lax.broadcast_to_rank# jax.lax.broadcast_to_rank(x, rank)[源代码][源代码]# 在 x 的维度前添加 1 以使其秩为 rank。 参数: x (ArrayLike) rank (int) 返回类型: Array
