跳转到主要内容

努力实现与Theano -> Aesara转换的向后兼容性

项目描述

aesara-theano-fallback

努力实现与Theano替换Aesara的向后兼容性,因为PyMC3项目。目标是提供一个几乎可以无缝替换导入aesara的库,当aesara未安装时,将回退到theano。这专门设计来支持exoplanetstarry项目,可能不支持所有您需要的功能。如果您发现不支持的功能,请提交拉取请求!

安装

可以使用pip安装此库

python -m pip install aesara-theano-fallback

使用方法

语法设计主要遵循aesara,因此以下内容通常可以工作

import aesara_theano_fallback.tensor as aet

对于顶级访问,使用

from aesara_theano_fallback import aesara

在Theano和Aesara之间,语法有一个显著的变化,即theano.gof模块被重命名为aesara.graph,内容也有所调整。对于exoplanet和starry,我们定义了一些自定义Op,您可以使用此库进行以下操作

from aesara_theano_fallback.graph import basic, op

class MyPythonOp(op.Op):
    def make_node(self, *args):
        # ...
        return basic.Apply(self, in_args, out_args)

class MyCOp(op.ExternalCOp):
    func_file = "./cpp_impl.cc"
    func_name = "APPLY_SPECIFIC(my_op_name)"
    # ...

项目详情


下载文件

下载适用于您的平台的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源代码分发

aesara_theano_fallback-0.1.0.tar.gz (6.5 kB 查看哈希值)

上传时间: 源代码

构建分发

aesara_theano_fallback-0.1.0-py3-none-any.whl (5.4 kB 查看哈希值)

上传时间: Python 3

由以下支持