ai4plasma
Getting Started
1. Installation Guide
2. Quick Start
3. Basic Concepts
4. Configuration
User Guide
1. Physics-Informed Machine Learning (PIML)
2. Neural Operators
3. Training, Logging, Checkpoints
4. Utilities
5. Plasma Models
Examples
1. Physics-Informed Machine Learning (PIML)
2. Operator Learning
3. Plasma Simulation
Core Modules
1. Core Module
2. Physics-Informed Machine Learning (PIML)
3. Operator Learning Module
4. Plasma Physics Module
5. Utilities
Development
1. Contributing
ai4plasma
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Y
A
act_fun (ai4plasma.core.network.CNN attribute)
(ai4plasma.core.network.FNN attribute)
add_equation() (ai4plasma.piml.pinn.PINN method)
ai4plasma.core.model
module
ai4plasma.core.network
module
ai4plasma.operator.deepcsnet
module
ai4plasma.operator.deeponet
module
ai4plasma.piml.cs_pinn
module
ai4plasma.piml.geo
module
ai4plasma.piml.meta_pinn
module
ai4plasma.piml.nas_pinn
module
ai4plasma.piml.pinn
module
ai4plasma.piml.rk_pinn
module
ai4plasma.plasma.arc
module
ai4plasma.plasma.prop
module
ai4plasma.utils.common
module
ai4plasma.utils.device
module
ai4plasma.utils.io
module
ai4plasma.utils.math
module
alpha() (ai4plasma.plasma.prop.CoronaPropSpline method)
arch_parameters() (ai4plasma.core.network.RelaxFNN method)
ArcPropSpline (class in ai4plasma.plasma.prop)
area (ai4plasma.piml.geo.GeoRect2D attribute)
B
backbone_net (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
BaseModel (class in ai4plasma.core.model)
bias_last (ai4plasma.operator.deepcsnet.DeepCSNet attribute)
(ai4plasma.operator.deeponet.DeepONet attribute)
bound (ai4plasma.piml.geo.GeoPoly2D attribute)
branch_inputs (ai4plasma.operator.deeponet.DeepONetDataset attribute)
branch_is_cnn (ai4plasma.operator.deeponet.DeepONet attribute)
branch_net (ai4plasma.operator.deeponet.DeepONet attribute)
build_up() (ai4plasma.core.network.RelaxFNN method)
C
C_in (ai4plasma.core.network.RelaxLayer attribute)
C_in_list (ai4plasma.core.network.RelaxFNN attribute)
calc_arc_cond() (ai4plasma.plasma.arc.StaArc1D method)
calc_GL_coefs() (in module ai4plasma.piml.cs_pinn)
calc_l2_err() (in module ai4plasma.utils.math)
calc_loss() (ai4plasma.core.model.BaseModel method)
(ai4plasma.operator.deepcsnet.DeepCSNetModel method)
(ai4plasma.operator.deeponet.DeepONetModel method)
(ai4plasma.piml.pinn.PINN method)
calc_relative_l2_err() (in module ai4plasma.utils.math)
cfg (ai4plasma.core.model.CfgBaseModel attribute)
CfgBaseModel (class in ai4plasma.core.model)
check_gpu() (in module ai4plasma.utils.device)
checkpoint_dir (ai4plasma.operator.deepcsnet.DeepCSNetModel attribute)
(ai4plasma.operator.deeponet.DeepONetModel attribute)
CNN (class in ai4plasma.core.network)
compute_loss() (ai4plasma.piml.meta_pinn.MetaTask method)
(ai4plasma.piml.meta_pinn.PINNTask method)
compute_residual() (ai4plasma.piml.pinn.EquationTerm method)
(ai4plasma.piml.pinn.PINN method)
conv_layers (ai4plasma.core.network.CNN attribute)
conv_net (ai4plasma.core.network.CNN attribute)
Corona1DRKModel (class in ai4plasma.piml.rk_pinn)
Corona1DRKNet (class in ai4plasma.piml.rk_pinn)
Corona1DRKVisCallback (class in ai4plasma.piml.rk_pinn)
CoronaPropSpline (class in ai4plasma.plasma.prop)
Cp() (ai4plasma.plasma.prop.ArcPropSpline method)
Cp_list (ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
create_dataloader() (ai4plasma.piml.pinn.EquationTerm method)
create_domain() (ai4plasma.piml.geo.Geo1D method)
(ai4plasma.piml.geo.Geometry method)
(ai4plasma.piml.geo.GeoPoly2D method)
(ai4plasma.piml.geo.GeoRect2D method)
(ai4plasma.piml.geo.GeoTime method)
create_lr_scheduler() (ai4plasma.piml.pinn.PINN method)
create_optimizer() (ai4plasma.piml.pinn.PINN method)
current() (ai4plasma.utils.common.Timer method)
D
D_e() (ai4plasma.plasma.prop.CoronaPropSpline method)
D_p() (ai4plasma.plasma.prop.CoronaPropSpline method)
dataloader (ai4plasma.operator.deepcsnet.DeepCSNetModel attribute)
(ai4plasma.operator.deeponet.DeepONetModel attribute)
dataset (ai4plasma.operator.deepcsnet.DeepCSNetModel attribute)
(ai4plasma.operator.deeponet.DeepONetModel attribute)
DeepCSNet (class in ai4plasma.operator.deepcsnet)
DeepCSNetDataset (class in ai4plasma.operator.deepcsnet)
DeepCSNetModel (class in ai4plasma.operator.deepcsnet)
DeepONet (class in ai4plasma.operator.deeponet)
DeepONetDataset (class in ai4plasma.operator.deeponet)
DeepONetModel (class in ai4plasma.operator.deeponet)
device (ai4plasma.utils.device.Device attribute)
Device (class in ai4plasma.utils.device)
device_id (ai4plasma.core.model.BaseModel attribute)
(ai4plasma.utils.device.Device attribute)
df_dX() (in module ai4plasma.utils.math)
do_after_each_epoch() (ai4plasma.core.model.CfgBaseModel method)
do_after_training() (ai4plasma.core.model.CfgBaseModel method)
do_before_training() (ai4plasma.core.model.CfgBaseModel method)
E
enable_adaptive_weights() (ai4plasma.piml.pinn.PINN method)
energy_inputs (ai4plasma.operator.deepcsnet.DeepCSNetDataset attribute)
energy_net (ai4plasma.operator.deepcsnet.DeepCSNet attribute)
epoch (ai4plasma.core.model.CfgBaseModel attribute)
epsilon (ai4plasma.piml.meta_pinn.MetaPINN attribute)
EquationTerm (class in ai4plasma.piml.pinn)
F
fc_layers_config (ai4plasma.core.network.CNN attribute)
fc_net (ai4plasma.core.network.CNN attribute)
fig_file_list (ai4plasma.core.model.CfgBaseModel attribute)
FNN (class in ai4plasma.core.network)
forward() (ai4plasma.core.network.CNN method)
(ai4plasma.core.network.FNN method)
(ai4plasma.core.network.Network method)
(ai4plasma.core.network.RelaxFNN method)
(ai4plasma.core.network.RelaxLayer method)
(ai4plasma.operator.deepcsnet.DeepCSNet method)
(ai4plasma.operator.deeponet.DeepONet method)
(ai4plasma.piml.cs_pinn.StaArc1DNet method)
(ai4plasma.piml.cs_pinn.TraArc1DNet method)
(ai4plasma.piml.cs_pinn.TraArc1DTempNet method)
(ai4plasma.piml.cs_pinn.TraArc1DVelNet method)
(ai4plasma.piml.meta_pinn.MetaStaArc1DNet method)
(ai4plasma.piml.rk_pinn.Corona1DRKNet method)
G
generate_mesh() (ai4plasma.plasma.arc.StaArc1D method)
geo (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
Geo1D (class in ai4plasma.piml.geo)
Geo1DTime (class in ai4plasma.piml.geo)
geo_space (ai4plasma.piml.geo.Geo1DTime attribute)
(ai4plasma.piml.geo.GeoPoly2DTime attribute)
geo_time (ai4plasma.piml.geo.Geo1DTime attribute)
(ai4plasma.piml.geo.GeoPoly2DTime attribute)
Geometry (class in ai4plasma.piml.geo)
GeoPoly2D (class in ai4plasma.piml.geo)
GeoPoly2DTime (class in ai4plasma.piml.geo)
GeoRect2D (class in ai4plasma.piml.geo)
GeoTime (class in ai4plasma.piml.geo)
get_bounding_box() (ai4plasma.piml.geo.GeoPoly2D method)
get_dataloader() (ai4plasma.piml.pinn.EquationTerm method)
get_equation() (ai4plasma.piml.pinn.PINN method)
get_equation_info() (ai4plasma.piml.pinn.PINN method)
get_feature_size() (ai4plasma.core.network.CNN method)
get_init_args() (ai4plasma.core.model.CfgBaseModel method)
get_json_args() (ai4plasma.core.model.CfgBaseModel method)
get_kwargs() (ai4plasma.core.model.CfgBaseModel method)
get_PhiNe_func_from_file() (in module ai4plasma.piml.rk_pinn)
get_query_data() (ai4plasma.piml.meta_pinn.MetaTask method)
get_support_data() (ai4plasma.piml.meta_pinn.MetaTask method)
get_task_id() (ai4plasma.piml.meta_pinn.MetaTask method)
get_Tfunc_from_file() (in module ai4plasma.piml.cs_pinn)
get_training_history() (ai4plasma.piml.pinn.PINN method)
get_TVfunc_from_file() (in module ai4plasma.piml.cs_pinn)
GL_degree (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
global_pool (ai4plasma.core.network.CNN attribute)
gs (ai4plasma.core.network.RelaxFNN attribute)
H
height (ai4plasma.piml.geo.GeoRect2D attribute)
history (ai4plasma.piml.meta_pinn.MetaPINN attribute)
(ai4plasma.piml.nas_pinn.NasPINN attribute)
I
I (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
,
[1]
(ai4plasma.plasma.arc.StaArc1D attribute)
(ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
img2gif() (in module ai4plasma.utils.io)
init_field() (ai4plasma.plasma.arc.TraArc1D method)
init_temp() (ai4plasma.plasma.arc.StaArc1D method)
init_weights() (ai4plasma.core.network.CNN method)
(ai4plasma.core.network.FNN method)
(ai4plasma.core.network.Network method)
(ai4plasma.core.network.RelaxFNN method)
(ai4plasma.core.network.RelaxLayer method)
inner_epochs (ai4plasma.piml.meta_pinn.MetaPINN attribute)
(ai4plasma.piml.nas_pinn.NasPINN attribute)
inner_lr (ai4plasma.piml.meta_pinn.MetaPINN attribute)
inner_opt (ai4plasma.piml.nas_pinn.NasPINN attribute)
input_dim (ai4plasma.core.network.CNN attribute)
interp_nec() (in module ai4plasma.plasma.prop)
interp_nec_log() (in module ai4plasma.plasma.prop)
interp_prop() (in module ai4plasma.plasma.prop)
interp_prop_log() (in module ai4plasma.plasma.prop)
interp_x() (in module ai4plasma.plasma.prop)
is_cnn_input (ai4plasma.operator.deeponet.DeepONetDataset attribute)
(ai4plasma.operator.deeponet.DeepONetModel attribute)
K
kappa() (ai4plasma.plasma.prop.ArcPropSpline method)
kappa_list (ai4plasma.plasma.arc.StaArc1D attribute)
(ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
L
last_epoch (ai4plasma.core.model.CfgBaseModel attribute)
last_outer_epochs (ai4plasma.piml.nas_pinn.NasPINN attribute)
layers (ai4plasma.core.network.FNN attribute)
(ai4plasma.core.network.RelaxFNN attribute)
LHS (ai4plasma.piml.geo.SamplingMode attribute)
,
[1]
linear_model() (ai4plasma.core.network.FNN method)
list2torch() (in module ai4plasma.utils.common)
load_butcher_table() (in module ai4plasma.piml.rk_pinn)
load_gs() (ai4plasma.core.network.RelaxFNN method)
load_last_epoch() (ai4plasma.core.model.CfgBaseModel method)
load_meta_model() (ai4plasma.piml.meta_pinn.MetaPINN method)
load_model() (ai4plasma.core.model.CfgBaseModel method)
load_model_from_file() (ai4plasma.core.model.CfgBaseModel method)
load_nas_model() (ai4plasma.piml.nas_pinn.NasPINN method)
load_optimizer() (ai4plasma.core.model.CfgBaseModel method)
load_weights() (ai4plasma.core.model.CfgBaseModel method)
loss_func (ai4plasma.core.model.BaseModel attribute)
(ai4plasma.operator.deepcsnet.DeepCSNetModel attribute)
(ai4plasma.piml.meta_pinn.MetaPINN attribute)
(ai4plasma.piml.meta_pinn.PINNTask attribute)
loss_list (ai4plasma.core.model.CfgBaseModel attribute)
lr_scheduler (ai4plasma.core.model.BaseModel attribute)
M
masks (ai4plasma.core.network.RelaxLayer attribute)
mesh (ai4plasma.plasma.arc.StaArc1D attribute)
(ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
meta_network (ai4plasma.piml.meta_pinn.MetaPINN attribute)
meta_test() (ai4plasma.piml.meta_pinn.MetaPINN method)
meta_train() (ai4plasma.piml.meta_pinn.MetaPINN method)
MetaPINN (class in ai4plasma.piml.meta_pinn)
MetaStaArc1DModel (class in ai4plasma.piml.meta_pinn)
MetaStaArc1DNet (class in ai4plasma.piml.meta_pinn)
MetaTask (class in ai4plasma.piml.meta_pinn)
module
ai4plasma.core.model
ai4plasma.core.network
ai4plasma.operator.deepcsnet
ai4plasma.operator.deeponet
ai4plasma.piml.cs_pinn
ai4plasma.piml.geo
ai4plasma.piml.meta_pinn
ai4plasma.piml.nas_pinn
ai4plasma.piml.pinn
ai4plasma.piml.rk_pinn
ai4plasma.plasma.arc
ai4plasma.plasma.prop
ai4plasma.utils.common
ai4plasma.utils.device
ai4plasma.utils.io
ai4plasma.utils.math
module (ai4plasma.operator.deepcsnet.DeepCSNet attribute)
molecule_inputs (ai4plasma.operator.deepcsnet.DeepCSNetDataset attribute)
molecule_net (ai4plasma.operator.deepcsnet.DeepCSNet attribute)
mu_e() (ai4plasma.plasma.prop.CoronaPropSpline method)
mu_p() (ai4plasma.plasma.prop.CoronaPropSpline method)
N
NasPINN (class in ai4plasma.piml.nas_pinn)
nec() (ai4plasma.plasma.prop.ArcPropSpline method)
nec_list (ai4plasma.plasma.arc.StaArc1D attribute)
(ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
net (ai4plasma.core.network.FNN attribute)
network (ai4plasma.core.model.BaseModel attribute)
(ai4plasma.core.network.RelaxFNN attribute)
(ai4plasma.operator.deepcsnet.DeepCSNetModel attribute)
(ai4plasma.operator.deeponet.DeepONetModel attribute)
(ai4plasma.piml.cs_pinn.StaArc1DNet attribute)
,
[1]
Network (class in ai4plasma.core.network)
neuron_list (ai4plasma.core.network.RelaxFNN attribute)
(ai4plasma.core.network.RelaxLayer attribute)
None (ai4plasma.core.network.Network attribute)
numpy2torch() (in module ai4plasma.utils.common)
O
op (ai4plasma.core.network.RelaxLayer attribute)
optimizer (ai4plasma.core.model.BaseModel attribute)
(ai4plasma.operator.deepcsnet.DeepCSNetModel attribute)
outer_epochs (ai4plasma.piml.meta_pinn.MetaPINN attribute)
(ai4plasma.piml.nas_pinn.NasPINN attribute)
outer_lr (ai4plasma.piml.meta_pinn.MetaPINN attribute)
outer_opt (ai4plasma.piml.nas_pinn.NasPINN attribute)
P
Parameters (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
(ai4plasma.piml.cs_pinn.StaArc1DNet attribute)
PINN (class in ai4plasma.piml.pinn)
pinn_model (ai4plasma.piml.meta_pinn.PINNTask attribute)
(ai4plasma.piml.meta_pinn.StaArc1DTask attribute)
(ai4plasma.piml.nas_pinn.NasPINN attribute)
PINNTask (class in ai4plasma.piml.meta_pinn)
plot_1d_comparison() (ai4plasma.piml.pinn.PINN static method)
plot_2d_comparison() (ai4plasma.piml.pinn.PINN static method)
plot_2d_heatmap() (ai4plasma.piml.pinn.PINN static method)
plot_error_heatmap() (ai4plasma.piml.pinn.PINN static method)
plot_residual_distribution() (ai4plasma.piml.pinn.PINN static method)
points (ai4plasma.piml.geo.GeoPoly2D attribute)
points_num (ai4plasma.piml.geo.GeoPoly2D attribute)
polygon (ai4plasma.piml.geo.GeoPoly2D attribute)
precision (ai4plasma.utils.math.Real attribute)
predict() (ai4plasma.core.model.BaseModel method)
(ai4plasma.operator.deepcsnet.DeepCSNetModel method)
(ai4plasma.operator.deeponet.DeepONetModel method)
(ai4plasma.piml.meta_pinn.MetaStaArc1DModel method)
(ai4plasma.piml.pinn.PINN method)
prepare_test_data() (ai4plasma.core.model.BaseModel method)
prepare_train_data() (ai4plasma.core.model.BaseModel method)
(ai4plasma.operator.deepcsnet.DeepCSNetModel method)
(ai4plasma.operator.deeponet.DeepONetModel method)
print_runing_time() (in module ai4plasma.utils.common)
prop (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
,
[1]
Q
query_data (ai4plasma.piml.meta_pinn.MetaTask attribute)
(ai4plasma.piml.meta_pinn.StaArc1DTask attribute)
R
R (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
,
[1]
(ai4plasma.piml.cs_pinn.StaArc1DNet attribute)
,
[1]
(ai4plasma.piml.meta_pinn.StaArc1DTask attribute)
(ai4plasma.plasma.arc.StaArc1D attribute)
(ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
RANDOM (ai4plasma.piml.geo.SamplingMode attribute)
,
[1]
read_json() (in module ai4plasma.utils.io)
read_nec_data() (in module ai4plasma.plasma.prop)
read_thermo_data() (in module ai4plasma.plasma.prop)
real (ai4plasma.utils.math.Real attribute)
Real (class in ai4plasma.utils.math)
register_visualization_callback() (ai4plasma.piml.meta_pinn.MetaPINN method)
(ai4plasma.piml.pinn.PINN method)
RelaxFNN (class in ai4plasma.core.network)
RelaxLayer (class in ai4plasma.core.network)
remove_equation() (ai4plasma.piml.pinn.PINN method)
rho() (ai4plasma.plasma.prop.ArcPropSpline method)
rho_list (ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
S
sample_all_domain() (ai4plasma.piml.geo.Geo1DTime method)
(ai4plasma.piml.geo.GeoPoly2DTime method)
sample_boundary() (ai4plasma.piml.geo.Geo1D method)
(ai4plasma.piml.geo.Geometry method)
(ai4plasma.piml.geo.GeoPoly2D method)
(ai4plasma.piml.geo.GeoRect2D method)
(ai4plasma.piml.geo.GeoTime method)
sample_domain() (ai4plasma.piml.geo.Geo1D method)
(ai4plasma.piml.geo.Geo1DTime method)
(ai4plasma.piml.geo.Geometry method)
(ai4plasma.piml.geo.GeoPoly2D method)
(ai4plasma.piml.geo.GeoRect2D method)
(ai4plasma.piml.geo.GeoTime method)
sample_mode (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
sample_space_time() (ai4plasma.piml.geo.GeoTime method)
sample_space_time_boundary() (ai4plasma.piml.geo.GeoTime method)
sample_space_time_list() (ai4plasma.piml.geo.Geo1DTime method)
(ai4plasma.piml.geo.GeoTime method)
SamplingMode (class in ai4plasma.piml.geo)
save_final_results() (ai4plasma.piml.cs_pinn.StaArc1DVisCallback method)
(ai4plasma.piml.cs_pinn.TraArc1DTempVisCallback method)
(ai4plasma.piml.cs_pinn.TraArc1DVisCallback method)
(ai4plasma.piml.rk_pinn.Corona1DRKVisCallback method)
save_gif() (ai4plasma.piml.cs_pinn.StaArc1DVisCallback method)
(ai4plasma.piml.cs_pinn.TraArc1DTempVisCallback method)
(ai4plasma.piml.cs_pinn.TraArc1DVisCallback method)
(ai4plasma.piml.rk_pinn.Corona1DRKVisCallback method)
save_meta_model() (ai4plasma.piml.meta_pinn.MetaPINN method)
save_nas_model() (ai4plasma.piml.nas_pinn.NasPINN method)
saved_model (ai4plasma.core.model.CfgBaseModel attribute)
search() (ai4plasma.piml.nas_pinn.NasPINN method)
searched_neuron() (ai4plasma.core.network.RelaxFNN method)
select_gpu_by_id() (in module ai4plasma.utils.device)
set_all_equation_weights() (ai4plasma.piml.pinn.PINN method)
set_boundary_field() (ai4plasma.plasma.arc.TraArc1D method)
set_boundary_temp() (ai4plasma.plasma.arc.StaArc1D method)
set_device() (ai4plasma.utils.device.Device method)
set_equation_data() (ai4plasma.piml.pinn.PINN method)
set_equation_weight() (ai4plasma.piml.pinn.PINN method)
set_float_precision() (ai4plasma.utils.math.Real method)
set_loss_func() (ai4plasma.core.model.BaseModel method)
set_lr_scheduler() (ai4plasma.core.model.BaseModel method)
set_optimizer() (ai4plasma.core.model.BaseModel method)
set_seed() (in module ai4plasma.utils.common)
set_solver() (ai4plasma.plasma.arc.StaArc1D method)
(ai4plasma.plasma.arc.TraArc1D method)
(ai4plasma.plasma.arc.TraArc1DNoV method)
set_torch_dtype() (ai4plasma.utils.math.Real method)
sigma() (ai4plasma.plasma.prop.ArcPropSpline method)
sigma_list (ai4plasma.plasma.arc.StaArc1D attribute)
(ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
solve() (ai4plasma.plasma.arc.StaArc1D method)
(ai4plasma.plasma.arc.TraArc1D method)
(ai4plasma.plasma.arc.TraArc1DNoV method)
solve_explicit() (ai4plasma.plasma.arc.TraArc1D method)
solve_onestep() (ai4plasma.plasma.arc.StaArc1D method)
(ai4plasma.plasma.arc.TraArc1D method)
(ai4plasma.plasma.arc.TraArc1DNoV method)
solver (ai4plasma.plasma.arc.StaArc1D attribute)
split_by_branch (ai4plasma.operator.deeponet.DeepONetDataset attribute)
split_by_mole (ai4plasma.operator.deepcsnet.DeepCSNetDataset attribute)
StaArc1D (class in ai4plasma.plasma.arc)
StaArc1DModel (class in ai4plasma.piml.cs_pinn)
StaArc1DNet (class in ai4plasma.piml.cs_pinn)
StaArc1DTask (class in ai4plasma.piml.meta_pinn)
StaArc1DVisCallback (class in ai4plasma.piml.cs_pinn)
start_epoch (ai4plasma.operator.deepcsnet.DeepCSNetModel attribute)
(ai4plasma.operator.deeponet.DeepONetModel attribute)
support_data (ai4plasma.piml.meta_pinn.MetaTask attribute)
(ai4plasma.piml.meta_pinn.StaArc1DTask attribute)
T
T (ai4plasma.plasma.arc.StaArc1D attribute)
(ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
T_red (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
,
[1]
targets (ai4plasma.operator.deepcsnet.DeepCSNetDataset attribute)
(ai4plasma.operator.deeponet.DeepONetDataset attribute)
task_id (ai4plasma.piml.meta_pinn.MetaTask attribute)
Tb (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
,
[1]
(ai4plasma.piml.cs_pinn.StaArc1DNet attribute)
,
[1]
te (ai4plasma.piml.geo.GeoTime attribute)
temp_list (ai4plasma.plasma.arc.StaArc1D attribute)
(ai4plasma.plasma.arc.TraArc1D attribute)
(ai4plasma.plasma.arc.TraArc1DNoV attribute)
test_data_size (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
test_tasks (ai4plasma.piml.meta_pinn.MetaPINN attribute)
Timer (class in ai4plasma.utils.common)
torch_device() (in module ai4plasma.utils.device)
total_epochs (ai4plasma.core.model.CfgBaseModel attribute)
TraArc1D (class in ai4plasma.plasma.arc)
TraArc1DModel (class in ai4plasma.piml.cs_pinn)
TraArc1DNet (class in ai4plasma.piml.cs_pinn)
TraArc1DNoV (class in ai4plasma.plasma.arc)
TraArc1DTempModel (class in ai4plasma.piml.cs_pinn)
TraArc1DTempNet (class in ai4plasma.piml.cs_pinn)
TraArc1DTempVisCallback (class in ai4plasma.piml.cs_pinn)
TraArc1DVelNet (class in ai4plasma.piml.cs_pinn)
TraArc1DVisCallback (class in ai4plasma.piml.cs_pinn)
train() (ai4plasma.core.model.BaseModel method)
(ai4plasma.operator.deepcsnet.DeepCSNetModel method)
(ai4plasma.operator.deeponet.DeepONetModel method)
(ai4plasma.piml.pinn.PINN method)
train_data_size (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
train_tasks (ai4plasma.piml.meta_pinn.MetaPINN attribute)
trunk_inputs (ai4plasma.operator.deepcsnet.DeepCSNetDataset attribute)
(ai4plasma.operator.deeponet.DeepONetDataset attribute)
trunk_net (ai4plasma.operator.deepcsnet.DeepCSNet attribute)
(ai4plasma.operator.deeponet.DeepONet attribute)
ts (ai4plasma.piml.geo.GeoTime attribute)
U
UNIFORM (ai4plasma.piml.geo.SamplingMode attribute)
,
[1]
update_data() (ai4plasma.piml.pinn.EquationTerm method)
update_weight() (ai4plasma.piml.pinn.EquationTerm method)
use_BN (ai4plasma.core.network.CNN attribute)
use_pooling (ai4plasma.core.network.CNN attribute)
V
V (ai4plasma.plasma.arc.TraArc1D attribute)
visualization_callback (ai4plasma.piml.nas_pinn.NasPINN attribute)
visualization_callbacks (ai4plasma.piml.meta_pinn.MetaPINN attribute)
VisualizationCallback (class in ai4plasma.piml.pinn)
visualize() (ai4plasma.piml.cs_pinn.StaArc1DVisCallback method)
(ai4plasma.piml.cs_pinn.TraArc1DTempVisCallback method)
(ai4plasma.piml.cs_pinn.TraArc1DVisCallback method)
(ai4plasma.piml.pinn.VisualizationCallback method)
(ai4plasma.piml.rk_pinn.Corona1DRKVisCallback method)
W
width (ai4plasma.piml.geo.GeoRect2D attribute)
Wq (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
writer (ai4plasma.core.model.CfgBaseModel attribute)
(ai4plasma.operator.deepcsnet.DeepCSNetModel attribute)
(ai4plasma.operator.deeponet.DeepONetModel attribute)
(ai4plasma.piml.meta_pinn.MetaPINN attribute)
(ai4plasma.piml.nas_pinn.NasPINN attribute)
X
xl (ai4plasma.piml.geo.Geo1D attribute)
xmax (ai4plasma.piml.geo.GeoRect2D attribute)
xmin (ai4plasma.piml.geo.GeoRect2D attribute)
Xq (ai4plasma.piml.cs_pinn.StaArc1DModel attribute)
xu (ai4plasma.piml.geo.Geo1D attribute)
Y
ymax (ai4plasma.piml.geo.GeoRect2D attribute)
ymin (ai4plasma.piml.geo.GeoRect2D attribute)