python - 模块 trax.fastmath 是否已弃用？

Question

TRAX 包的模块 trax.fastmath 是否已弃用？我正在使用此模块，但返回 ModuleNotFoundError。

Answer 1

我用 trax.math 替换了 trax.fastmath，一切正常。我仍然不确定 Trax 文档中是否对此进行了更新。