Jax, Numpy & Scipy
发布日期:2022-02-14 23:02:57 浏览次数:122 分类:技术文章

本文共 155 字,大约阅读时间需要 1 分钟。

Google开源的Jax据说很厉害,主要是用来解决numpy在GPU上支持度不太友好的开源包(,,等均同此),今天小试一下,确实有优秀的地方

测试(仅限很少部分的numpy api,文档中有Scipy的api)

  • Version

    在这里插入图片描述

  • 随机数 & jax替换numpy的点乘

    在这里插入图片描述

  • jit

    在这里插入图片描述
    在这里插入图片描述

  • 函数

    在这里插入图片描述

  • 梯度计算

    在这里插入图片描述

Reference

转载地址:https://blog.csdn.net/fish2009122/article/details/113826936 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:Elasticsearch operation by python
下一篇:建模之常见优化器(Keras)

发表评论

最新留言

很好
[***.229.124.182]2024年04月25日 11时19分21秒