folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

PBC Layer (#252)

Browse Source 
* update docs/tests
* tutorial and device fix

---------

Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.local>
Co-authored-by: Dario Coscia <dariocoscia@dhcp-015.eduroam.sissa.it>
Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.lan>
Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.Home>
This commit is contained in:
Dario Coscia
2024-03-01 18:15:45 +01:00
committed by GitHub
parent c92a2832d5
commit 4cfd90b904
13 changed files with 984 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/model/layers/__init__.py
View File

@@ -9,6 +9,7 @@ __all__ = [
"FourierBlock2D",
"FourierBlock3D",
"PODBlock",
"PeriodicBoundaryEmbedding"
]
from .convolution_2d import ContinuousConvBlock
@@ -20,3 +21,4 @@ from .spectral import (
)
from .fourier import FourierBlock1D, FourierBlock2D, FourierBlock3D
from .pod import PODBlock
from .embedding import PeriodicBoundaryEmbedding