12 lines
244 B
Python
12 lines
244 B
Python
""" 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):
|
|
|
|
|