jax.lax.broadcast_shapes# jax.lax.broadcast_shapes(*shapes: tuple[int, ...]) → tuple[int, ...][原始碼]# jax.lax.broadcast_shapes(*shapes: tuple[int | core.Tracer, ...]) → tuple[int | core.Tracer, ...] 傳回從 NumPy 廣播 shapes 產生的形狀。
