MCPcopy Create free account

hub / github.com/rtqichen/torchdiffeq / functions

Functions279 in github.com/rtqichen/torchdiffeq

⨍ Functions 279◇ Types & classes 82
Methodevent_fn
(t, state)
examples/learn_physics.py:66
Methodf
(t, x)
tests/odeint_tests.py:181
Methodf
(t, x)
tests/odeint_tests.py:212
Methodf
(t, x)
tests/norm_tests.py:44
Functionfind_tensor_attributes
(module)
torchdiffeq/_impl/adjoint.py:233
Methodforward
(self, t, y)
tests/problems.py:14
Methodforward
(self, t, y)
tests/problems.py:22
Methodforward
(self, t, y)
tests/problems.py:43
Methodforward
(self, t, y)
tests/problems.py:57
Methodforward
(self, t, x)
tests/odeint_tests.py:281
Methodforward
(self, t, y)
tests/gradient_tests.py:98
Methodforward
(self, t, x)
tests/norm_tests.py:34
Methodforward
(ctx, scratch, value, index)
torchdiffeq/_impl/rk_common.py:33
Methodforward
(self, t, y)
torchdiffeq/_impl/misc.py:143
Methodforward
(self, t, y)
torchdiffeq/_impl/misc.py:154
Methodforward
(self, t, y)
torchdiffeq/_impl/misc.py:164
Methodforward
(self, t, y, *, perturb=Perturb.NONE)
torchdiffeq/_impl/misc.py:180
Methodforward
(ctx, x1, out)
torchdiffeq/_impl/misc.py:350
Methodforward
(ctx, shapes, func, y0, t, rtol, atol, method, options, event_fn, adjoint_rtol, adjoint_atol, adjoint_method,
torchdiffeq/_impl/adjoint.py:11
Methodforward
event_t is the solution to event_fn
torchdiffeq/_impl/odeint.py:200
Methodforward
(self, t, state)
examples/bouncing_ball.py:25
Methodforward
(self, x)
examples/odenet_mnist.py:60
Methodforward
(self, t, x)
examples/odenet_mnist.py:86
Methodforward
(self, t, x)
examples/odenet_mnist.py:104
Methodforward
(self, x)
examples/odenet_mnist.py:123
Methodforward
(self, x)
examples/odenet_mnist.py:142
Methodforward
(self, t, state)
examples/learn_physics.py:20
Methodforward
(self, t, state)
examples/learn_physics.py:35
Methodforward
(self, t, state)
examples/learn_physics.py:46
Methodforward
(self, x)
examples/learn_physics.py:132
Methodforward
(self, t, state)
examples/learn_physics.py:157
Methodforward
(self, t, y)
examples/ode_demo.py:36
Methodforward
(self, t, y)
examples/ode_demo.py:127
Methodforward
(self, t, x)
examples/latent_ode.py:118
Methodforward
(self, x, h)
examples/latent_ode.py:137
Methodforward
(self, t, states)
examples/cnf.py:45
Methodforward
(self, t)
examples/cnf.py:97
Methodinit
(m)
examples/learn_physics.py:149
Methodintegrate
(self, t)
torchdiffeq/_impl/scipy_wrapper.py:27
Methodintegrate
(self, t)
torchdiffeq/_impl/solvers.py:28
Methodintegrate_until_event
(self, t0, event_fn)
torchdiffeq/_impl/solvers.py:44
Methodlarge_norm
(tensor)
tests/norm_tests.py:249
Functionlearning_rate_fn
(itr)
examples/odenet_mnist.py:217
Methodmake_spy_on_adjoint_norm
(adjoint_norm, actual_norm)
tests/norm_tests.py:97
Methodnfe
(self)
examples/odenet_mnist.py:129
Functionnp_func
(t, y)
torchdiffeq/_impl/scipy_wrapper.py:53
Functionodeint_adjoint
(func, y0, t, *, rtol=1e-7, atol=1e-9, method=None, options=None, event_fn=None, adjoint_rt
torchdiffeq/_impl/adjoint.py:156
Functionodeint_dense
(func, y0, t0, t1, *, rtol=1e-7, atol=1e-9, method=None, options=None)
torchdiffeq/_impl/odeint.py:111
Functionrk4_step_func
(func, t0, dt, t1, y0, f0=None, perturb=False)
torchdiffeq/_impl/rk_common.py:99
Methodsimulate
(self, times)
examples/learn_physics.py:61
Methodspy_on_adjoint_norm
(tensor)
tests/norm_tests.py:100
Methodtest_adjoint
(self)
tests/odeint_tests.py:60
Methodtest_adjoint
(self)
tests/event_tests.py:51
Methodtest_adjoint
(self)
tests/gradient_tests.py:25
Methodtest_adjoint_against_odeint
Test against dopri5
tests/gradient_tests.py:34
Methodtest_adjoint_norm
(self)
tests/norm_tests.py:91
Methodtest_against_dopri5
(self)
tests/gradient_tests.py:106
Methodtest_forward
(self)
tests/api_tests.py:12
Methodtest_gradient
(self)
tests/api_tests.py:28
Methodtest_grid_constructor
(self)
tests/odeint_tests.py:211
Methodtest_large_norm
(self)
tests/norm_tests.py:244
Methodtest_min_max_step
(self)
tests/odeint_tests.py:252
Methodtest_norm
(self)
tests/norm_tests.py:43
Methodtest_odeint
(self)
tests/odeint_tests.py:17
Methodtest_odeint
(self)
tests/odeint_tests.py:79
Methodtest_odeint
(self)
tests/odeint_tests.py:99
Methodtest_odeint
(self)
tests/event_tests.py:14
Methodtest_odeint
(self)
tests/gradient_tests.py:13
Methodtest_odeint_jump_t
(self)
tests/odeint_tests.py:127
Methodtest_odeint_perturb
(self)
tests/odeint_tests.py:164
Methodtest_seminorm
(self)
tests/norm_tests.py:272
Methodtest_steps
(self)
tests/odeint_tests.py:310
Methodtest_wrong_callback
(self)
tests/odeint_tests.py:290
Methodvalid_callbacks
(cls)
torchdiffeq/_impl/scipy_wrapper.py:47
Methodvalid_callbacks
(cls)
torchdiffeq/_impl/rk_common.py:208
Methodvalid_callbacks
(cls)
torchdiffeq/_impl/solvers.py:25
Methody_exact
(self, t)
tests/problems.py:17
Methody_exact
(self, t)
tests/problems.py:25
Methody_exact
(self, t)
tests/problems.py:47
← previous201–279 of 279, ranked by callers