jax.numpy.shape¶. jax.numpy. shape (a)¶. Return the shape of an array. Parameters. a (array_like) – Input array. Returns. shape – The elements of the shape ...
確定! 回上一頁