jax.lax
共有5个方法/函数/属性,点击链接查看相应的源代码示例。
1. jax.lax.scan() ,6个项目使用
2. jax.lax.conv_general_dilated() ,5个项目使用
3. jax.lax.max() ,5个项目使用
4. jax.lax.reduce_window() ,5个项目使用
5. jax.lax.stop_gradient() ,5个项目使用
注:本文中的示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。