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-20 13:19:17 +00:00
committed by Nicola Demo
parent 98f7694d6f
commit 6bc8d81a5c
1 changed files with 20 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
pina/model/avno.py
View File

@@ -62,19 +62,26 @@ class AveragingNeuralOperator(KernelNeuralOperator):
# check hidden dimensions match
input_lifting_net = next(lifting_net.parameters()).size()[-1]
output_lifting_net = lifting_net(
torch.rand(size=next(lifting_net.parameters()).size())
).shape[-1]
projecting_net_input=next(projecting_net.parameters()).size()[-1]
torch.rand(size=next(lifting_net.parameters()).size())
).shape[-1]
projecting_net_input = next(projecting_net.parameters()).size()[-1]
if len(field_indices)+len(coordinates_indices) != input_lifting_net:
raise ValueError('The lifting_net must take as input the '
'coordinates vector and the field vector.')
if output_lifting_net+len(coordinates_indices) != projecting_net_input:
raise ValueError('The projecting_net input must be equal to'
'the embedding dimension (which is the output) '
'of the lifting_net plus the dimension of the '
'coordinates, i.e. len(coordinates_indices).')
if len(field_indices) + len(coordinates_indices) != input_lifting_net:
raise ValueError(
"The lifting_net must take as input the "
"coordinates vector and the field vector."
)
if (
output_lifting_net + len(coordinates_indices)
!= projecting_net_input
):
raise ValueError(
"The projecting_net input must be equal to"
"the embedding dimension (which is the output) "
"of the lifting_net plus the dimension of the "
"coordinates, i.e. len(coordinates_indices)."
)
# assign
self.coordinates_indices = coordinates_indices
@@ -108,4 +115,4 @@ class AveragingNeuralOperator(KernelNeuralOperator):
new_batch = self._integral_kernels(new_batch)
new_batch = concatenate((new_batch, points_tmp), dim=2)
new_batch = self._projection_operator(new_batch)
return new_batch
return new_batch