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, ...] 返回对 shapes 进行 NumPy 广播后得到的形状。
