Files
PINA/pina/model/layers/spectral.py
Dario Coscia 15ecaacb7c Layers and Models update PR
* add residual block
* add test conv and residual block
* modify FFN kwargs
2023-11-17 09:51:29 +01:00

16 lines
270 B
Python

import torch
import torch.nn as nn
from ...utils import check_consistency
class SpectralConvBlock(nn.Module):
"""
Implementation of spectral convolution block.
"""
def __init__(self):
super().__init__()
def forward(self, x):
pass