aTEAM-master

所属分类:其他
开发工具:Perl
文件大小:50KB
下载次数:0
上传日期:2021-02-24 04:36:42
上 传 者partohaghighi
说明:  A pyTorch Extension for Applied Mathematics

文件列表:
LICENSE (1068, 2020-03-17)
__init__.py (87, 2020-03-17)
nn (0, 2020-03-17)
nn\__init__.py (47, 2020-03-17)
nn\functional (0, 2020-03-17)
nn\functional\__init__.py (50, 2020-03-17)
nn\functional\interpolation.py (6697, 2020-03-17)
nn\functional\utils.py (3117, 2020-03-17)
nn\modules (0, 2020-03-17)
nn\modules\FD.py (9110, 2020-03-17)
nn\modules\Interpolation.py (6422, 2020-03-17)
nn\modules\MK.py (3100, 2020-03-17)
nn\modules\__init__.py (65, 2020-03-17)
optim (0, 2020-03-17)
optim\NFI.py (8762, 2020-03-17)
optim\PGManager.py (11963, 2020-03-17)
optim\__init__.py (218, 2020-03-17)
pdetools (0, 2020-03-17)
pdetools\__init__.py (131, 2020-03-17)
pdetools\example (0, 2020-03-17)
pdetools\example\__init__.py (20, 2020-03-17)
pdetools\example\burgers2d.py (7171, 2020-03-17)
pdetools\example\cde2d.py (8789, 2020-03-17)
pdetools\example\cdr2d.py (5442, 2020-03-17)
pdetools\example\rd2d.py (5207, 2020-03-17)
pdetools\init.py (1532, 2020-03-17)
pdetools\spectral.py (4730, 2020-03-17)
pdetools\stepper.py (4659, 2020-03-17)
pdetools\upwind.py (2691, 2020-03-17)
test (0, 2020-03-17)
test\fd_test.py (2738, 2020-03-17)
test\fdproj_test.py (325, 2020-03-17)
test\interp_test2d.py (3702, 2020-03-17)
test\interp_test3d.py (4428, 2020-03-17)
test\mk_test.py (1455, 2020-03-17)
test\modules_test2d.py (3463, 2020-03-17)
test\modules_test3d.py (6228, 2020-03-17)
test\optim_quickstart.py (8139, 2020-03-17)
... ...

# aTEAM **A** py**T**orch **E**xtension for **A**pplied **M**athematics This version is compatible with pytorch (1.0.1) and later. You can create a conda environment for pytorch1: ``` conda create -n torch1 python=3 jupyter source activate torch1 conda install pytorch=1 torchvision cudatoolkit=9.2 -c pytorch # or conda install pytorch-cpu=1 -c pytorch ``` ## Some code maybe useful to you (News: add optim QuickStart) - aTEAM.optim.NumpyFuntionInterface: This function enable us to optimize pytorch modules with external optimizer such as scipy.optimize.lbfgsb.fmin_l_bfgs_b, see test/optim_quickstart.py - aTEAM.nn.modules.MK: [Moment matrix](https://arxiv.org/abs/1710.09668) & convolution kernel convertor: aTEAM.nn.modules.MK.M2K, aTEAM.nn.module.MK.K2M - aTEAM.nn.modules.Interpolation: Lagrange interpolation in a n-dimensional box: aTEAM.nn.modules.Interpolation.LagrangeInterp, aTEAM.nn.modules.Interpolation.LagrangeInterpFixInputs - aTEAM.nn.functional.utils.tensordot: It is similar to numpy.tensordot For more usages pls refer to aTEAM/test/*.py # PDE-Net aTEAM is a basic library for PDE-Net & PDE-Net 2.0[(source code)](https://github.com/ZichaoLong/PDE-Net): - [PDE-Net: Learning PDEs from Data](https://arxiv.org/abs/1710.09668)[(ICML 2018)](https://icml.cc/Conferences/2018)
[Long Zichao](https://scholar.google.com/citations?user=0KXcwnkAAAAJ&hl=zh-CN), [Lu Yiping](https://web.stanford.edu/~yplu/), [Ma Xianzhong](https://www.researchgate.net/profile/Xianzhong_Ma), [Dong Bin](http://bicmr.pku.edu.cn/~dongbin) - [PDE-Net 2.0: Learning PDEs from Data with A Numeric-Symbolic Hybrid Deep Network](https://arxiv.org/abs/1812.04426)
[Long Zichao](https://scholar.google.com/citations?user=0KXcwnkAAAAJ&hl=zh-CN), [Lu Yiping](https://web.stanford.edu/~yplu/), [Dong Bin](http://bicmr.pku.edu.cn/~dongbin) If you find this code useful for your research then please cite ``` @inproceedings{long2018pdeI, title={PDE-Net: Learning PDEs from Data}, author={Long, Zichao and Lu, Yiping and Ma, Xianzhong and Dong, Bin}, booktitle={International Conference on Machine Learning}, pages={3214--3222}, year={2018} } @article{long2018pdeII, title={PDE-Net 2.0: Learning PDEs from Data with A Numeric-Symbolic Hybrid Deep Network}, author={Long, Zichao and Lu, Yiping and Dong, Bin}, journal={arXiv preprint arXiv:1812.04426}, year={2018} } ```

近期下载者

相关文件


收藏者