jax.enable_x64

jax.enable_x64 = <jax._src.config.State 对象>

用于 jax_enable_x64 配置选项的上下文管理器。

启用 64 位类型的使用

参数:

new_val (任意类型)