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

🎨 Format Python code with psf/black

Browse Source
This commit is contained in:
ndem0
2024-03-05 11:31:14 +00:00
committed by Nicola Demo
parent 43f69242ab
commit 3d72205380
2 changed files with 25 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
pina/model/avno.py
View File

@@ -70,11 +70,22 @@ class AveragingNeuralOperator(KernelNeuralOperator):
self.coordinates_indices = coordinates_indices
self.field_indices = field_indices
integral_net = nn.Sequential(
*[AVNOBlock(inner_size, func) for _ in range(n_layers)])
lifting_net = FeedForward(dimension + input_numb_fields, inner_size,
inner_size, n_layers, func)
projection_net = FeedForward(inner_size + dimension, output_numb_fields,
inner_size, n_layers, func)
*[AVNOBlock(inner_size, func) for _ in range(n_layers)]
)
lifting_net = FeedForward(
dimension + input_numb_fields,
inner_size,
inner_size,
n_layers,
func,
)
projection_net = FeedForward(
inner_size + dimension,
output_numb_fields,
inner_size,
n_layers,
func,
)
super().__init__(lifting_net, integral_net, projection_net)
def forward(self, x):