lasdi.latent_space

Attributes

act_dict

Classes

`MultiLayerPerceptron`
`Autoencoder`

Functions

initial_condition_latent(param_grid, physics, autoencoder)

Outputs the initial condition in the latent space: Z0 = encoder(U0)

Module Contents

lasdi.latent_space.act_dict

lasdi.latent_space.initial_condition_latent(param_grid, physics, autoencoder): Outputs the initial condition in the latent space: Z0 = encoder(U0)

class lasdi.latent_space.MultiLayerPerceptron(layer_sizes, act_type='sigmoid', reshape_index=None, reshape_shape=None, threshold=0.1, value=0.0, num_heads=1)

Bases: torch.nn.Module

n_layers

layer_sizes

fcs = []

reshape_index

reshape_shape

act_type

use_multihead = False

forward(x)

apply_attention(x, act_idx)

init_weight()

class lasdi.latent_space.Autoencoder(physics, config)

Bases: torch.nn.Module

qgrid_size

space_dim

hidden_units

n_z

layer_sizes

act_type

threshold

value

num_heads

encoder

decoder

forward(x)

export()

load(dict_)