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

Fourier Block and minor fixes

Browse Source 
* Adding fourier block 1d/2d/3d
* Adding docs to SpectralConvBlock1D/2D/3D and to FourierBlock1D/2D/3D
* Adding tests for fourier block
This commit is contained in:
Dario Coscia
2023-09-07 18:18:28 +02:00
committed by Nicola Demo
parent 2bf42d5fea
commit 83ecdb0eab
4 changed files with 228 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,9 +3,13 @@ __all__ = [
'ResidualBlock',
'SpectralConvBlock1D',
'SpectralConvBlock2D',
'SpectralConvBlock3D'
'SpectralConvBlock3D',
'FourierBlock1D',
'FourierBlock2D',
'FourierBlock3D',
]
from .convolution_2d import ContinuousConvBlock
from .residual import ResidualBlock
from .spectral import SpectralConvBlock1D, SpectralConvBlock2D, SpectralConvBlock3D
from .fourier import FourierBlock1D, FourierBlock2D, FourierBlock3D