jax.lax.broadcast_形状# jax.lax.broadcast_shapes(*shapes: tuple[int, ...]) → tuple[int, ...][源代码][源代码]# jax.lax.broadcast_shapes(*shapes: tuple[int | core.Tracer, ...]) → tuple[int | core.Tracer, ...] 返回由NumPy广播 shapes 得到的结果形状。
