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对象进行传参调用,可以说这二者在使用特性上是存在一致性的。




posted on 2024-01-10 08:49  Angry_Panda  阅读(25)  评论(0)    收藏  举报

导航