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

Rename nabla -> laplacian

Browse Source
This commit is contained in:
Pasquale Africa
2023-07-20 08:15:13 +00:00
committed by Nicola Demo
parent 92e0e4920b
commit 6c627c70e3
16 changed files with 62 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/problems/parametric_elliptic_optimal_control_alpha_variable.py
View File

@@ -3,7 +3,7 @@ import torch
from pina import Span, Condition
from pina.problem import SpatialProblem, ParametricProblem
from pina.operators import grad, nabla
from pina.operators import grad, laplacian
# ===================================================== #
# #
@@ -42,11 +42,11 @@ class ParametricEllipticOptimalControl(SpatialProblem, ParametricProblem):
# equation terms as in https://arxiv.org/pdf/2110.13530.pdf
def term1(input_, output_):
laplace_p = nabla(output_, input_, components=['p'], d=['x1', 'x2'])
laplace_p = laplacian(output_, input_, components=['p'], d=['x1', 'x2'])
return output_.extract(['y']) - input_.extract(['mu']) - laplace_p
def term2(input_, output_):
laplace_y = nabla(output_, input_, components=['y'], d=['x1', 'x2'])
laplace_y = laplacian(output_, input_, components=['y'], d=['x1', 'x2'])
return - laplace_y - output_.extract(['u_param'])
def state_dirichlet(input_, output_):

4
examples/problems/parametric_poisson.py
View File

@@ -1,7 +1,7 @@
import torch
from pina.problem import SpatialProblem, ParametricProblem
from pina.operators import nabla
from pina.operators import laplacian
from pina import Span, Condition
# ===================================================== #
@@ -28,7 +28,7 @@ class ParametricPoisson(SpatialProblem, ParametricProblem):
force_term = torch.exp(
- 2*(input_.extract(['x']) - input_.extract(['mu1']))**2
- 2*(input_.extract(['y']) - input_.extract(['mu2']))**2)
return nabla(output_.extract(['u']), input_) - force_term
return laplacian(output_.extract(['u']), input_) - force_term
# define nill dirichlet boundary conditions
def nil_dirichlet(input_, output_):

6
examples/problems/poisson.py
View File

@@ -2,7 +2,7 @@ import numpy as np
import torch
from pina.problem import SpatialProblem
from pina.operators import nabla
from pina.operators import laplacian
from pina import Condition, Span
# ===================================================== #
@@ -26,8 +26,8 @@ class Poisson(SpatialProblem):
def laplace_equation(input_, output_):
force_term = (torch.sin(input_.extract(['x'])*torch.pi) *
torch.sin(input_.extract(['y'])*torch.pi))
nabla_u = nabla(output_.extract(['u']), input_)
return nabla_u - force_term
delta_u = laplacian(output_.extract(['u']), input_)
return delta_u - force_term
# define nill dirichlet boundary conditions
def nil_dirichlet(input_, output_):

11
examples/problems/stokes.py
View File

@@ -2,7 +2,7 @@ import numpy as np
import torch
from pina.problem import SpatialProblem
from pina.operators import nabla, grad, div
from pina.operators import laplacian, grad, div
from pina import Condition, Span, LabelTensor
# ===================================================== #
@@ -25,11 +25,10 @@ class Stokes(SpatialProblem):
# define the momentum equation
def momentum(input_, output_):
nabla_ = torch.hstack((LabelTensor(nabla(output_.extract(['ux']), input_), ['x']),
LabelTensor(nabla(output_.extract(['uy']), input_), ['y'])))
return - nabla_ + grad(output_.extract(['p']), input_)
# define the continuity equation
delta_ = torch.hstack((LabelTensor(laplacian(output_.extract(['ux']), input_), ['x']),
LabelTensor(laplacian(output_.extract(['uy']), input_), ['y'])))
return - delta_ + grad(output_.extract(['p']), input_)
def continuity(input_, output_):
return div(output_.extract(['ux', 'uy']), input_)