difftorch:PyTorch的差异化API

时间:2024-04-09 02:08:56
【文件属性】:

文件名称:difftorch:PyTorch的差异化API

文件大小:13KB

文件格式:ZIP

更新时间:2024-04-09 02:08:56

Python

difftorch difftorch为功能分化API 。它实现的操作反映了可用的那些操作。 安装 安装最新的软件包 要使用可用的最新版本,请运行: pip install difftorch 从源安装 要从源代码安装,请克隆此存储库并使用以下命令安装difftorch软件包: git clone https://github.com/gbaydin/difftorch.git cd difftorch pip install . 文献资料 一些示例函数演示了这些操作: import torch import difftorch def f_rosenbrock ( x ): x , y = x [ 0 ], x [ 1 ] return ( 1. - x ) ** 2 + 100. * ( y - x ** 2 ) ** 2 def f_spherical_cartes


【文件预览】:
difftorch-master
----.travis.yml(203B)
----difftorch()
--------difftorch.py(11KB)
--------util.py(2KB)
--------__init__.py(497B)
----tests()
--------test_generic_api.py(4KB)
--------test_api_short.py(6KB)
--------test_api.py(8KB)
--------test_util.py(679B)
--------run_all.sh(31B)
----LICENSE(1KB)
----setup.py(1KB)
----.gitignore(2KB)
----README.md(6KB)
----notebooks()
--------examples.ipynb(5KB)

网友评论