Skip to main content

为 NumPy 带来插入式自动微分

项目描述

MyGrad 是一个轻量级库,它为 NumPy 添加了自动微分功能——它唯一的依赖是 NumPy!它的主要目标是使自动微分在 Python/NumPy 生态系统中易于访问和使用。

MyGrad 引入了一个张量对象,其行为类似于 NumPy 的 ndarray 对象,但它构建了一个计算图,使 MyGrad 能够执行反向模式微分(即“反向传播”)。通过利用 NumPy 的 ufunc/函数覆盖机制,MyGrad 的张量与 NumPy 的数学函数套件“本机”工作,因此它们可以链接在一起形成可微分的计算图。

NumPy 用于广播操作、生成数组视图、执行就地操作以及允许对数组进行“基本”和“高级”索引的系统都得到 MyGrad 的高保真支持。

项目详情


下载文件

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

源分布

mygrad-2.1.0.tar.gz (148.8 kB 查看哈希)

已上传 source

内置分布

mygrad-2.1.0-py3-none-any.whl (170.5 kB 查看哈希

已上传 py3