603f56d264
* Building FNO for 1D/2D/3D data * Fixing bug in trunk/branch net in DeepONEt * Fixing type check bug in spectral conv * Adding tests for FNO * Fixing bug in Fourier Layer (conv1d/2d/3d)
12 lines
224 B
Python
12 lines
224 B
Python
__all__ = [
|
|
'FeedForward',
|
|
'MultiFeedForward',
|
|
'DeepONet',
|
|
'FNO',
|
|
]
|
|
|
|
from .feed_forward import FeedForward
|
|
from .multi_feed_forward import MultiFeedForward
|
|
from .deeponet import DeepONet
|
|
from .fno import FNO
|