Jax框架的Traced object特性与TensorFlow的placeholder的一致性
前文:
Jax框架的static与Traced Operations —— Static vs Traced Operations
前文讨论分析了Jax的static特性和Traced特性,这些谈下个人对jax的Traced特性的一些个人看法。
我个人认为Jax的底层组件使用的是和TensorFlow一致的组件,因此其上层的使用也必然存在一定的相似性。虽然TensorFlow因为其API的复杂性以及前端调用的复杂性被逐步放弃,但是其底层的高性能计算的后端还是比较优秀的,毕竟是Google的全公司之力打造出的计算后端,使用了当前最新的计算技术和最高水平的编译技术。
而jax的Traced特性和TensorFlow的placeholder也是极为相似的,也都是在进行编译时起到占位的一个作用。在编译的时候都只需要指定Traced对象和placeholder对象的shape和type即可,编译后的代码均可以用符合编译时指定的shape和type的Traced对象进行传参调用,可以说这二者在使用特性上是存在一致性的。
本博客是博主个人学习时的一些记录,不保证是为原创,个别文章加入了转载的源地址,还有个别文章是汇总网上多份资料所成,在这之中也必有疏漏未加标注处,如有侵权请与博主联系。
如果未特殊标注则为原创,遵循 CC 4.0 BY-SA 版权协议。
posted on 2024-01-10 08:49 Angry_Panda 阅读(25) 评论(0) 收藏 举报
浙公网安备 33010602011771号