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

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_)