jax.experimental.enable_x64

jax.experimental.enable_x64(new_val=True)

實驗性上下文管理器，用於暫時啟用 X64 模式。

用法

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

另請參閱

jax.experimental.enable_x64

暫時啟用 X64 模式。

參數:

new_val (bool)