Index

A

append() (neural_operators.mcmc.tracer.Tracer method) apply_grid_vector_dirichlet_bc() (in module neural_operators.pde.meshUtilities)

assemble() (neural_operators.pde.pdeModel.PDEModel method) (neural_operators.prior.priorSampler.PriorSampler method)

B

boundaryU() (neural_operators.pde.pdeModel.PDEModel static method)

build_vector_vertex_maps() (in module neural_operators.pde.fenicsUtilities)

C

compl_mul2d() (neural_operators.nn.fno.torch_fno2dlayer.FNO2DLayer method) compute_mean() (neural_operators.pde.pdeModel.PDEModel method) (neural_operators.prior.priorSampler.PriorSampler method) compute_sample_errors() (in module neural_operators.mcmc.compute_sample_errors) compute_svd() (neural_operators.data.dataMethods.DataProcessor method)

convert_np_to_tensor() (neural_operators.data.dataMethods.DataHandler method) (neural_operators.nn.deeponet.torch_deeponet.DeepONet method) (neural_operators.nn.fno.torch_fno2d.FNO2D method) (neural_operators.nn.pcanet.torch_pcanet.PCANet method) CorrectedSurrogateModel (class in neural_operators.mcmc.correctedSurrogateModel)

D

DataHandler (class in neural_operators.data.dataMethods) DataProcessor (class in neural_operators.data.dataMethods) DataProcessorFNO (class in neural_operators.data.dataMethods) DataProcessorTF (class in neural_operators.data.dataMethods) decoder_X() (neural_operators.data.dataMethods.DataProcessor method) (neural_operators.data.dataMethods.DataProcessorFNO method) (neural_operators.data.dataMethods.DataProcessorTF method)

decoder_Y() (neural_operators.data.dataMethods.DataProcessor method) (neural_operators.data.dataMethods.DataProcessorFNO method) (neural_operators.data.dataMethods.DataProcessorTF method) DeepONet (class in neural_operators.nn.deeponet.torch_deeponet)

E

empty_m() (neural_operators.pde.pdeModel.PDEModel method) empty_sample() (neural_operators.prior.priorSampler.PriorSampler method) empty_u() (neural_operators.pde.pdeModel.PDEModel method) encoder_X() (neural_operators.data.dataMethods.DataProcessor method) (neural_operators.data.dataMethods.DataProcessorFNO method) (neural_operators.data.dataMethods.DataProcessorTF method)

encoder_Y() (neural_operators.data.dataMethods.DataProcessor method) (neural_operators.data.dataMethods.DataProcessorFNO method) (neural_operators.data.dataMethods.DataProcessorTF method) eval_at_vertices() (in module neural_operators.pde.fenicsUtilities)

F

field_plot() (in module neural_operators.plotting.field_plot) field_plot_grid() (in module neural_operators.plotting.field_plot) FNO2D (class in neural_operators.nn.fno.torch_fno2d) FNO2DLayer (class in neural_operators.nn.fno.torch_fno2dlayer) forward() (neural_operators.nn.deeponet.torch_deeponet.DeepONet method) (neural_operators.nn.fno.torch_fno2d.FNO2D method) (neural_operators.nn.fno.torch_fno2dlayer.FNO2DLayer method) (neural_operators.nn.mlp.torch_mlp.MLP method) (neural_operators.nn.pcanet.torch_pcanet.PCANet method)

fourier_transform() (neural_operators.nn.fno.torch_fno2dlayer.FNO2DLayer method) function_to_vector() (in module neural_operators.pde.fenicsUtilities) (neural_operators.pde.pdeModel.PDEModel method) (neural_operators.prior.priorSampler.PriorSampler method) function_to_vertex() (in module neural_operators.pde.fenicsUtilities) (neural_operators.pde.pdeModel.PDEModel method) (neural_operators.prior.priorSampler.PriorSampler method)

G

get_data_to_save() (neural_operators.data.dataMethods.DataProcessor method) (neural_operators.data.dataMethods.DataProcessorFNO method) (neural_operators.data.dataMethods.DataProcessorTF method) get_default_plot_mix_collection_data() (in module neural_operators.plotting.plot_mix_collection)

get_dirichlet_bc() (in module neural_operators.pde.meshUtilities) get_grid_dirichlet_bc() (in module neural_operators.pde.meshUtilities) get_mcmc_params_for_tracer() (neural_operators.mcmc.mcmc.MCMC method) get_surrogate_specs() (in module neural_operators.nn.nn_util) grid_to_nodes_u() (neural_operators.mcmc.surrogateModel.SurrogateModelFNO method)

I

init() (neural_operators.mcmc.tracer.Tracer method) init_logger() (neural_operators.mcmc.mcmc.MCMC method)

interpolate_mixed_vector_to_grid() (in module neural_operators.pde.meshUtilities) is_point_on_dirichlet_boundary() (neural_operators.pde.pdeModel.PDEModel static method)

L

lift_SVD() (neural_operators.data.dataMethods.DataProcessor method) linear_transform() (neural_operators.nn.fno.torch_fno2dlayer.FNO2DLayer method) load_data_and_deeponet() (in module neural_operators.nn.deeponet.load_data_and_deeponet) load_data_and_fno() (in module neural_operators.nn.fno.load_data_and_fno) load_data_and_model() (in module neural_operators.nn.nn_util) load_data_and_pcanet() (in module neural_operators.nn.pcanet.load_data_and_pcanet) load_surrogate_model() (in module neural_operators.nn.nn_util) load_X_data() (neural_operators.data.dataMethods.DataProcessor method) (neural_operators.data.dataMethods.DataProcessorFNO method)

load_Y_data() (neural_operators.data.dataMethods.DataProcessor method) (neural_operators.data.dataMethods.DataProcessorFNO method) locate_index() (in module neural_operators.plotting.plot_svd) logger() (neural_operators.mcmc.mcmc.MCMC method) logLikelihood() (neural_operators.mcmc.mcmc.MCMC method) logPosterior() (neural_operators.mcmc.mcmc.MCMC method) logPrior() (neural_operators.prior.priorSampler.PriorSampler method)

M

MCMC (class in neural_operators.mcmc.mcmc) mcmc_plot_fields() (in module neural_operators.mcmc.mcmc_plot_fields) mcmc_plot_fields_base() (in module neural_operators.mcmc.mcmc_plot_fields) mesh_cells() (in module neural_operators.pde.meshUtilities) minibatch() (neural_operators.data.dataMethods.DataProcessorTF method) MLP (class in neural_operators.nn.mlp.torch_mlp) module neural_operators.data.dataMethods neural_operators.mcmc.compute_sample_errors neural_operators.mcmc.correctedSurrogateModel neural_operators.mcmc.mcmc neural_operators.mcmc.mcmc_plot_fields neural_operators.mcmc.mcmc_postprocess neural_operators.mcmc.state neural_operators.mcmc.surrogateModel neural_operators.mcmc.tracer neural_operators.nn.deeponet.load_data_and_deeponet neural_operators.nn.deeponet.torch_deeponet neural_operators.nn.fno.load_data_and_fno neural_operators.nn.fno.torch_fno2d neural_operators.nn.fno.torch_fno2dlayer neural_operators.nn.mlp.torch_mlp neural_operators.nn.nn_util neural_operators.nn.pcanet.load_data_and_pcanet neural_operators.nn.pcanet.torch_pcanet neural_operators.pde.fenicsUtilities neural_operators.pde.meshUtilities neural_operators.pde.pdeModel neural_operators.plotting.field_plot neural_operators.plotting.plot_curve neural_operators.plotting.plot_loss neural_operators.plotting.plot_mix_collection neural_operators.plotting.plot_svd neural_operators.plotting.point_plot neural_operators.prior.priorSampler

N

neural_operators.data.dataMethods module neural_operators.mcmc.compute_sample_errors module neural_operators.mcmc.correctedSurrogateModel module neural_operators.mcmc.mcmc module neural_operators.mcmc.mcmc_plot_fields module neural_operators.mcmc.mcmc_postprocess module neural_operators.mcmc.state module neural_operators.mcmc.surrogateModel module neural_operators.mcmc.tracer module neural_operators.nn.deeponet.load_data_and_deeponet module neural_operators.nn.deeponet.torch_deeponet module neural_operators.nn.fno.load_data_and_fno module neural_operators.nn.fno.torch_fno2d module neural_operators.nn.fno.torch_fno2dlayer module

neural_operators.nn.mlp.torch_mlp module neural_operators.nn.nn_util module neural_operators.nn.pcanet.load_data_and_pcanet module neural_operators.nn.pcanet.torch_pcanet module neural_operators.pde.fenicsUtilities module neural_operators.pde.meshUtilities module neural_operators.pde.pdeModel module neural_operators.plotting.field_plot module neural_operators.plotting.plot_curve module neural_operators.plotting.plot_loss module neural_operators.plotting.plot_mix_collection module neural_operators.plotting.plot_svd module neural_operators.plotting.point_plot module neural_operators.prior.priorSampler module nodes_to_grid_m() (neural_operators.mcmc.surrogateModel.SurrogateModelFNO method)

P

PCANet (class in neural_operators.nn.pcanet.torch_pcanet) PDEModel (class in neural_operators.pde.pdeModel) plot() (neural_operators.mcmc.mcmc.MCMC method) plot_collection() (in module neural_operators.plotting.plot_mix_collection) plot_collection_grid() (in module neural_operators.plotting.plot_mix_collection) plot_curve() (in module neural_operators.plotting.plot_curve) plot_loss() (in module neural_operators.plotting.plot_loss) plot_mcmc_tracer_results() (in module neural_operators.mcmc.mcmc_postprocess) plot_mix_collection() (in module neural_operators.plotting.plot_mix_collection)

plot_s_vec_values() (in module neural_operators.plotting.plot_svd) point_plot() (in module neural_operators.plotting.point_plot) predict() (neural_operators.nn.deeponet.torch_deeponet.DeepONet method) (neural_operators.nn.fno.torch_fno2d.FNO2D method) (neural_operators.nn.pcanet.torch_pcanet.PCANet method) predict_from_m() (neural_operators.mcmc.surrogateModel.SurrogateModel method) PriorSampler (class in neural_operators.prior.priorSampler) process_and_print() (neural_operators.mcmc.mcmc.MCMC method) project_SVD() (neural_operators.data.dataMethods.DataProcessor method) proposal() (neural_operators.mcmc.mcmc.MCMC method)

Q

quick_field_plot() (in module neural_operators.plotting.field_plot)

quick_field_plot_grid() (in module neural_operators.plotting.field_plot) quick_point_plot() (in module neural_operators.plotting.point_plot)

R

run() (neural_operators.mcmc.mcmc.MCMC method)

S

sample() (neural_operators.mcmc.mcmc.MCMC method) samplePrior() (neural_operators.pde.pdeModel.PDEModel method) save() (neural_operators.mcmc.mcmc.MCMC method) (neural_operators.mcmc.tracer.Tracer method) set() (neural_operators.mcmc.state.State method) set_diffusivity() (neural_operators.prior.priorSampler.PriorSampler method) solveFwd() (neural_operators.mcmc.correctedSurrogateModel.CorrectedSurrogateModel method) (neural_operators.mcmc.mcmc.MCMC method) (neural_operators.mcmc.surrogateModel.SurrogateModel method) (neural_operators.mcmc.surrogateModel.SurrogateModelFNO method) (neural_operators.pde.pdeModel.PDEModel method)

State (class in neural_operators.mcmc.state) state_to_obs() (neural_operators.mcmc.mcmc.MCMC method) stats() (neural_operators.mcmc.tracer.Tracer method) SurrogateModel (class in neural_operators.mcmc.surrogateModel) SurrogateModelFNO (class in neural_operators.mcmc.surrogateModel)

T

test_dirichlet_bc_functions() (in module neural_operators.pde.meshUtilities) testbatch() (neural_operators.data.dataMethods.DataProcessorTF method) Tracer (class in neural_operators.mcmc.tracer)

train() (neural_operators.nn.deeponet.torch_deeponet.DeepONet method) (neural_operators.nn.fno.torch_fno2d.FNO2D method) (neural_operators.nn.pcanet.torch_pcanet.PCANet method)

V

vector_to_function() (in module neural_operators.pde.fenicsUtilities) (neural_operators.pde.pdeModel.PDEModel method) (neural_operators.prior.priorSampler.PriorSampler method) vector_to_vertex() (in module neural_operators.pde.fenicsUtilities)

vector_values_at_nodes() (in module neural_operators.plotting.field_plot) vertex_to_function() (in module neural_operators.pde.fenicsUtilities) (neural_operators.pde.pdeModel.PDEModel method) (neural_operators.prior.priorSampler.PriorSampler method) vertex_to_vector() (in module neural_operators.pde.fenicsUtilities)

W

write_mesh_xdmf() (in module neural_operators.pde.meshUtilities)