jax.experimental.disable_x64

jax.experimental.disable_x64()

实验性上下文管理器，用于临时禁用 X64 模式。

警告

此上下文管理器自 JAX v0.8.0 起已弃用，并将在 JAX v0.9.0 中移除。请改用 jax.enable_x64()。

用法

>>> x = np.arange(5, dtype='float64')
>>> with _disable_x64():
...   print(jnp.asarray(x).dtype)
...
float32

另请参阅

jax.experimental.enable_x64

临时启用 X64 模式。