refact
This commit is contained in:
11
pina/model/layers/messa_passing.py
Normal file
11
pina/model/layers/messa_passing.py
Normal 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):
|
||||
|
||||
|
||||
Reference in New Issue
Block a user