即将离开知乎

您即将离开知乎，请注意您的账号和财产安全。

https://jiayiwu.me/blog/2021/04/05/learning-about-jax-axes-in-vmap.html