Question about calculation of elem in DarcyDataset

scaomath / galerkin-transformer

[NeurIPS 2021] Galerkin Transformer: a linear attention without softmax for Partial Differential Equations

MIT License

214 stars 28 forks source link

Question about calculation of elem in DarcyDataset #1

Closed liyang-7 closed 2 years ago

liyang-7 commented 2 years ago

Nice job. Is there any reference for calculating the elem matrix in DarcyDataset (libs/ft.py line650-661)? What needs to be done to generalize from 2D to 3D

scaomath commented 2 years ago

I am implementing a pure PyTorch finite element package (use ML to speed up traditional FEM computation not the current data fitting problem everyone is doing). The elem was an artifact of that (kinda lazy back in May and June and did not clean up the repo from my developing repo). It is not used in the actual computation just for visualization with plotly.graph_objects.Surface which needs a triangulation input. Let me clean the repo a bit after this month.

liyang-7 commented 2 years ago

Yea, the elem matrix is not used in the network computation. Thanks a lot