如何在Google-Jax中使用梯度卷积?

感谢阅读我的问题! 我刚刚开始学习Jax中的自定义梯度…