Physics-Informed Machine Learning (PIML)

julia

Transolver(;
    out_dim::Int,
    nheads::Int=8,
    func_dim::Union{Int,Nothing}=nothing,
    spatial_dim::Union{Int,Nothing}=nothing,
    preprocess::Union{Nothing,AbstractLuxLayer}=nothing,
    geometry_kind::Symbol=:unstructured,
    temporal_input::Bool=false,
    hidden_dim::Int=128,
    activation=NNlib.gelu,
    num_layers::Int=4,
    nheads::Int,
    hidden_dim::Int,
    dropout_rate::AbstractFloat=0.0f0,
    activation=NNlib.gelu,
    mlp_ratio::Int=4,
    last_layer::Bool=false,
    slice_num::Int=32,
    shapelist=nothing,
)

Implements the Transolver model from [3]. Currently we only support unstructured meshes without temporal input. Internally uses PhysicsSelfAttentionIrregularMesh.

source

Trusted by

Physics-Informed Machine Learning (PIML) ​

Physics-Informed Machine Learning (PIML)