df673cad4e
* solvers -> solver * adaptive_functions -> adaptive_function * callbacks -> callback * operators -> operator * pinns -> physics_informed_solver * layers -> block
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):
|
|
|
|
|