jax.Array.shape

abstract property Array.shape: tuple[int, ...]

数组的形状。