Why Gemfury?
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Debian packages
RPM packages
NuGet packages
emaballarin / neuraloperator python
Repository URL to install this package:
|
Version:
2.0.0 ▾
|
"""
SFNO - Spherical Fourier Neural Operator
Replaces the default SpectralConv (a convolution in the frequency domain
over Fourier basis functions) with a SphericalConv (a convolution over the
spherical harmonic basis functions)
"""
from ..layers.spherical_convolution import SphericalConv
from .fno import FNO, partialclass
SFNO = partialclass("SFNO", FNO, factorization="dense", conv_module=SphericalConv)
SFNO.__doc__ = SFNO.__doc__.replace("Fourier", "Spherical Fourier", 1)
SFNO.__doc__ = SFNO.__doc__.replace("FNO", "SFNO")
SFNO.__doc__ = SFNO.__doc__.replace("fno", "sfno")