A convolutional neural network based Poisson solver
poisson_CNN is a convolutional neural network model whih estimates the solution of the Poisson equation with four Dirichlet boundary conditions on rectangular grids of variable sizes.
An article describing the performance of our model is available on arXiv. A separate repository of Jupyter notebooks to generate the results outlined in the article is available.
Installation requires CUDA set up to work with tensorflow-gpu version 2.0.0 or newer. To install, please simply run
pip install poisson_CNN
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Hashes for poisson_CNN-0.2.0-py3-none-any.whl