idrlnet

• idrlnet package
  • use_cpu()
  • use_gpu()
  • Subpackages
    • idrlnet.architecture package
      • Submodules
      • idrlnet.architecture.grid module
      • idrlnet.architecture.layer module
      • idrlnet.architecture.mlp module
    • idrlnet.geo_utils package
      • Submodules
      • idrlnet.geo_utils.geo module
      • idrlnet.geo_utils.geo_builder module
      • idrlnet.geo_utils.geo_obj module
      • idrlnet.geo_utils.sympy_np module
    • idrlnet.pde_op package
      • Submodules
      • idrlnet.pde_op.equations module
      • idrlnet.pde_op.operator module
  • Submodules
  • idrlnet.callbacks module
    • GradientReceiver
      • GradientReceiver.receive_notify()
    • HandleResultReceiver
      • HandleResultReceiver.receive_notify()
    • SummaryReceiver
      • SummaryReceiver.receive_notify()
  • idrlnet.data module
    • DataNode
      • DataNode.counter
      • DataNode.lambda_outputs
      • DataNode.loss_fn
      • DataNode.sample()
      • DataNode.sample_fn
      • DataNode.sigma
    • SampleDomain
      • SampleDomain.sampling()
    • datanode()
    • get_data_node()
    • get_data_nodes()
  • idrlnet.graph module
    • Vertex
      • Vertex.counter
    • VertexTaskPipeline
      • VertexTaskPipeline.MAX_STACK_ALLOWED
      • VertexTaskPipeline.display()
      • VertexTaskPipeline.evaluation_order_list
      • VertexTaskPipeline.forward_pipeline()
      • VertexTaskPipeline.operation_order()
      • VertexTaskPipeline.to_json()
  • idrlnet.header module
    • TestFun
      • TestFun.registered
      • TestFun.run()
    • testmemo()
  • idrlnet.net module
    • NetNode
      • NetNode.counter
      • NetNode.fixed
      • NetNode.is_reference
      • NetNode.load_state_dict()
      • NetNode.net
      • NetNode.require_no_grad
      • NetNode.state_dict()
  • idrlnet.node module
    • Node
      • Node.derivatives
      • Node.evaluate
      • Node.inputs
      • Node.name
      • Node.new_node()
      • Node.outputs
  • idrlnet.optim module
    • Optimizable
      • Optimizable.OPTIMIZER_MAP
      • Optimizable.SCHEDULE_MAP
      • Optimizable.configure_optimizers()
      • Optimizable.optimizers
      • Optimizable.parse_configure()
      • Optimizable.parse_lr_schedule()
      • Optimizable.parse_optimizer()
      • Optimizable.schedulers
    • get_available_class()
  • idrlnet.pde module
    • ExpressionNode
    • PdeNode
      • PdeNode.equations
      • PdeNode.make_nodes()
      • PdeNode.sub_nodes
      • PdeNode.suffix
  • idrlnet.receivers module
    • Notifier
      • Notifier.notify()
      • Notifier.receivers
      • Notifier.register_receiver()
    • Receiver
      • Receiver.receive_notify()
    • Signal
      • Signal.AFTER_COMPUTE_LOSS
      • Signal.BEFORE_BACKWARD
      • Signal.BEFORE_COMPUTE_LOSS
      • Signal.REGISTER
      • Signal.SOLVE_END
      • Signal.SOLVE_START
      • Signal.TRAIN_PIPE_END
      • Signal.TRAIN_PIPE_START
  • idrlnet.shortcut module
  • idrlnet.solver module
    • Solver
      • Solver.append_sample_domain()
      • Solver.compute_loss()
      • Solver.configure_optimizers()
      • Solver.forward_through_all_graph()
      • Solver.generate_computation_pipeline()
      • Solver.generate_in_out_dict()
      • Solver.get_domain_parameter()
      • Solver.get_sample_domain()
      • Solver.infer_step()
      • Solver.init_load()
      • Solver.load()
      • Solver.network_dir
      • Solver.sample_domains
      • Solver.sample_variables_from_domains()
      • Solver.save()
      • Solver.set_domain_parameter()
      • Solver.set_param_ranges()
      • Solver.solve()
      • Solver.summary_receiver
      • Solver.train_pipe()
      • Solver.trainable_parameters
  • idrlnet.torch_util module
    • integral
      • integral.default_assumptions
      • integral.name
    • torch_lambdify()
  • idrlnet.variable module
    • Loss
      • Loss.Identity
      • Loss.L1
      • Loss.square
    • Variables
      • Variables.cat()
      • Variables.differentiate_()
      • Variables.differentiate_one_step_()
      • Variables.from_tensor()
      • Variables.merge_tensor()
      • Variables.save()
      • Variables.subset()
      • Variables.to_csv()
      • Variables.to_dataframe()
      • Variables.to_ndarray()
      • Variables.to_ndarray_()
      • Variables.to_torch_tensor_()
      • Variables.to_vtu()
      • Variables.var_differentiate_one_step()
      • Variables.weighted_loss()
    • export_var()