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

Spectral Convolution Addition

Browse Source 
* Implementing 1D/2D/3D spectral conv
* Implementing tests for 1D/2D/3d spectral conv
This commit is contained in:
Dario Coscia
2023-09-07 17:13:51 +02:00
committed by Nicola Demo
parent 190ee0561d
commit b7f9694cb9
10 changed files with 367 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,7 +1,11 @@
__all__ = [
'ContinuousConvBlock',
'ResidualBlock'
'ResidualBlock',
'SpectralConvBlock1D',
'SpectralConvBlock2D',
'SpectralConvBlock3D'
]
from .convolution_2d import ContinuousConvBlock
from .residual import ResidualBlock
from .spectral import SpectralConvBlock1D, SpectralConvBlock2D, SpectralConvBlock3D