This commit is contained in:
Nicola Demo
2024-09-09 10:50:54 +02:00
parent 9d9c2aa23e
commit f0d68b34c7
23 changed files with 480 additions and 229 deletions

View File

@@ -0,0 +1,11 @@
""" Module for Averaging Neural Operator Layer class. """
from torch import nn, mean
from torch_geometric.nn import MessagePassing, InstanceNorm, radius_graph
from pina.utils import check_consistency
class MessagePassingBlock(nn.Module):