首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

JAX 中文文档(五)

(int32[2,15*b]),) 与 TensorFlow 的互操作 原文:jax.readthedocs.io/en/latest/export/jax2tf.html 参见JAX2TF 文档...return jnp.minimum(x, 0) >>> print(func(2)) 0 要更深入了解关于跟踪器与常规值、具体值与抽象值之间的微妙差别,您可能需要阅读关于不同类型 JAX 值的文档...>>> func(0) Array(1, dtype=int32, weak_type=True) 要更深入理解跟踪器与常规值以及具体与抽象值之间的微妙差别,您可以阅读有关不同类型 JAX 值的文档...在本文档中,我们描述了 Pallas,这是 JAX 的一个扩展,可以使用类似 Triton 的模型为 GPU 和 TPU 编写核心程序。...屏蔽的操作语义可以由编译器决定(如果我们正确理解文档的话,Triton 在掩码时避免从内存读取/写入)。

18610
领券