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

in progress

Browse Source
This commit is contained in:
Nicola Demo
2024-06-18 11:20:58 +02:00
parent e4c69515dd
commit 2b71e0148d
7 changed files with 534 additions and 364 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/model/fno.py
View File

@@ -4,7 +4,7 @@ Fourier Neural Operator Module.
import torch
import torch.nn as nn
from pina import LabelTensor
from ..label_tensor import LabelTensor
import warnings
from ..utils import check_consistency
from .layers.fourier import FourierBlock1D, FourierBlock2D, FourierBlock3D