MCPcopy Create free account

hub / github.com/rtqichen/torchdiffeq / types & classes

Types & classes82 in github.com/rtqichen/torchdiffeq

⨍ Functions 279◇ Types & classes 82
↓ 3 callersClassBouncingBallExample
examples/bouncing_ball.py:14
↓ 3 callersClassResBlock
examples/odenet_mnist.py:48
↓ 3 callersClassRunningAverageMeter
Computes and stores the average and current value
examples/odenet_mnist.py:147
↓ 3 callersClass_NeuralF
tests/odeint_tests.py:271
↓ 3 callersClass_NeuralF
tests/norm_tests.py:21
↓ 2 callersClassConcatConv2d
examples/odenet_mnist.py:76
↓ 2 callersClassRunningAverageMeter
Computes and stores the average and current value
examples/ode_demo.py:131
↓ 2 callersClass_JumpF
tests/odeint_tests.py:114
↓ 2 callersClass_ReverseFunc
torchdiffeq/_impl/misc.py:158
↓ 1 callersClassCNF
Adapted from the NumPy implementation at: https://gist.github.com/rtqichen/91924063aa4cc95e7ef30b3a5491cc52
examples/cnf.py:34
↓ 1 callersClassDecoder
examples/latent_ode.py:147
↓ 1 callersClassDopri5Solver
torchdiffeq/_impl/dopri5.py:33
↓ 1 callersClassEventFn
examples/learn_physics.py:27
↓ 1 callersClassFlatten
examples/odenet_mnist.py:137
↓ 1 callersClassHamiltonianDynamics
examples/learn_physics.py:14
↓ 1 callersClassHyperNetwork
Hyper-network allowing f(z(t), t) to change with time. Adapted from the NumPy implementation at: https://gist.github.com/rtqichen/91924063aa4
examples/cnf.py:77
↓ 1 callersClassInstantaneousStateChange
examples/learn_physics.py:41
↓ 1 callersClassLambda
examples/ode_demo.py:34
↓ 1 callersClassLatentODEfunc
examples/latent_ode.py:108
↓ 1 callersClassNFEDiffEq
tests/DETEST/run.py:11
↓ 1 callersClassNeuralODE
examples/learn_physics.py:136
↓ 1 callersClassNeuralPhysics
examples/learn_physics.py:52
↓ 1 callersClassODEBlock
examples/odenet_mnist.py:116
↓ 1 callersClassODEFunc
examples/ode_demo.py:111
↓ 1 callersClassODEfunc
examples/odenet_mnist.py:92
↓ 1 callersClassOdefunc
tests/gradient_tests.py:92
↓ 1 callersClassRecognitionRNN
examples/latent_ode.py:128
↓ 1 callersClassRunningAverageMeter
Computes and stores the average and current value
examples/latent_ode.py:162
↓ 1 callersClassRunningAverageMeter
Computes and stores the average and current value
examples/cnf.py:117
↓ 1 callersClassSineODE
tests/problems.py:21
↓ 1 callersClass_PerturbFunc
torchdiffeq/_impl/misc.py:174
↓ 1 callersClass_TupleFunc
torchdiffeq/_impl/misc.py:137
↓ 1 callersClass_TupleInputOnlyFunc
torchdiffeq/_impl/misc.py:148
ClassAdamsBashforth
torchdiffeq/_impl/fixed_adams.py:226
ClassAdamsBashforthMoulton
torchdiffeq/_impl/fixed_adams.py:164
ClassAdaptiveHeunSolver
torchdiffeq/_impl/adaptive_heun.py:22
ClassAdaptiveStepsizeEventODESolver
torchdiffeq/_impl/solvers.py:38
ClassAdaptiveStepsizeODESolver
torchdiffeq/_impl/solvers.py:7
ClassBosh3Solver
torchdiffeq/_impl/bosh3.py:19
ClassConstantODE
tests/problems.py:7
ClassDopri8Solver
torchdiffeq/_impl/dopri8.py:73
ClassEuler
torchdiffeq/_impl/fixed_grid.py:6
ClassExpODE
tests/problems.py:56
ClassFehlberg2
torchdiffeq/_impl/fehlberg2.py:19
ClassFixedGridDIRKODESolver
torchdiffeq/_impl/rk_common.py:486
ClassFixedGridFIRKODESolver
torchdiffeq/_impl/rk_common.py:378
ClassFixedGridODESolver
torchdiffeq/_impl/solvers.py:52
ClassGaussLegendre4
torchdiffeq/_impl/fixed_grid_implicit.py:62
ClassGaussLegendre6
torchdiffeq/_impl/fixed_grid_implicit.py:77
ClassHeun2
torchdiffeq/_impl/fixed_grid.py:48
ClassHeun3
torchdiffeq/_impl/fixed_grid.py:32
ClassImplicitEuler
torchdiffeq/_impl/fixed_grid_implicit.py:19
ClassImplicitFnGradientRerouting
torchdiffeq/_impl/odeint.py:197
ClassImplicitMidpoint
torchdiffeq/_impl/fixed_grid_implicit.py:33
ClassLinearODE
tests/problems.py:31
ClassMidpoint
torchdiffeq/_impl/fixed_grid.py:14
ClassOdeintAdjointMethod
torchdiffeq/_impl/adjoint.py:8
ClassPerturb
torchdiffeq/_impl/misc.py:168
ClassRK4
torchdiffeq/_impl/fixed_grid.py:24
ClassRKAdaptiveStepsizeODESolver
torchdiffeq/_impl/rk_common.py:161
ClassRadauIIA3
torchdiffeq/_impl/fixed_grid_implicit.py:91
ClassRadauIIA5
torchdiffeq/_impl/fixed_grid_implicit.py:106
ClassSDIRK2
torchdiffeq/_impl/fixed_grid_implicit.py:121
ClassScipyWrapperODESolver
torchdiffeq/_impl/scipy_wrapper.py:7
ClassSine
examples/learn_physics.py:131
ClassTRBDF2
torchdiffeq/_impl/fixed_grid_implicit.py:138
ClassTestCallbacks
tests/odeint_tests.py:289
ClassTestCollectionState
tests/api_tests.py:11
ClassTestCompareAdjointGradient
tests/gradient_tests.py:89
ClassTestDiscontinuities
tests/odeint_tests.py:126
ClassTestEventHandling
tests/event_tests.py:12
ClassTestGradient
tests/gradient_tests.py:12
ClassTestGridConstructor
tests/odeint_tests.py:210
ClassTestMinMaxStep
tests/odeint_tests.py:251
ClassTestNoIntegration
tests/odeint_tests.py:98
ClassTestNorms
tests/norm_tests.py:42
ClassTestScipySolvers
tests/odeint_tests.py:77
ClassTestSolverError
tests/odeint_tests.py:15
ClassTrapezoid
torchdiffeq/_impl/fixed_grid_implicit.py:57
ClassTsit5Solver
torchdiffeq/_impl/tsit5.py:79
Class_StitchGradient
torchdiffeq/_impl/misc.py:348
Class_UncheckedAssign
torchdiffeq/_impl/rk_common.py:31