MCPcopy Create free account

hub / github.com/rtqichen/torchdiffeq / functions

Functions279 in github.com/rtqichen/torchdiffeq

⨍ Functions 279◇ Types & classes 82
↓ 1 callersMethodreset
(self)
examples/ode_demo.py:138
↓ 1 callersMethodreset
(self)
examples/latent_ode.py:169
↓ 1 callersMethodreset
(self)
examples/cnf.py:124
↓ 1 callersFunctionrk2_step_func
butcher_tableu should be of the form [ [0 , 0 , 0 ], [c_2, a_{21}, 0 ], [0 , b_1 , b_2 ], ] ht
torchdiffeq/_impl/rk_common.py:142
↓ 1 callersFunctionrk3_step_func
butcher_tableu should be of the form [ [0 , 0 , 0 , 0], [c_2, a_{21}, 0 , 0], [c_3, a_{31}, a_{32},
torchdiffeq/_impl/rk_common.py:121
↓ 1 callersFunctionset_learning_rate
(optimizer, lr)
examples/learn_physics.py:207
↓ 1 callersMethodsimulate
(self, nbounces=1)
examples/bouncing_ball.py:73
↓ 1 callersFunctiontrace_df_dz
Calculates the trace of the Jacobian df/dz. Stolen from: https://github.com/rtqichen/ffjord/blob/master/lib/layers/odefunc.py#L13
examples/cnf.py:66
↓ 1 callersMethodupdate
(self, val)
examples/latent_ode.py:173
↓ 1 callersMethodupdate
(self, val)
examples/cnf.py:128
↓ 1 callersMethodvalid_callbacks
(cls)
torchdiffeq/_impl/solvers.py:82
↓ 1 callersFunctionvisualize
(true_y, pred_y, odefunc, itr)
examples/ode_demo.py:67
FunctionA1
()
tests/DETEST/detest.py:8
FunctionA2
()
tests/DETEST/detest.py:15
FunctionA3
()
tests/DETEST/detest.py:22
FunctionA4
()
tests/DETEST/detest.py:29
FunctionA5
()
tests/DETEST/detest.py:36
FunctionB1
()
tests/DETEST/detest.py:45
FunctionB2
()
tests/DETEST/detest.py:58
FunctionB3
()
tests/DETEST/detest.py:72
FunctionB4
()
tests/DETEST/detest.py:86
FunctionB5
()
tests/DETEST/detest.py:101
FunctionC1
()
tests/DETEST/detest.py:118
FunctionC2
()
tests/DETEST/detest.py:135
FunctionC3
()
tests/DETEST/detest.py:152
FunctionC4
()
tests/DETEST/detest.py:170
FunctionC5
()
tests/DETEST/detest.py:188
FunctionE1
()
tests/DETEST/detest.py:263
FunctionE2
()
tests/DETEST/detest.py:276
FunctionE3
()
tests/DETEST/detest.py:289
FunctionE4
()
tests/DETEST/detest.py:302
FunctionE5
()
tests/DETEST/detest.py:315
Method__call__
(self, t, x)
tests/odeint_tests.py:118
Method__call__
(self, t, y)
tests/DETEST/run.py:17
Method__init__
(self)
tests/problems.py:9
Method__init__
(self, width, oscillate)
tests/odeint_tests.py:272
Method__init__
(self)
tests/gradient_tests.py:93
Method__init__
(self, width, oscillate)
tests/norm_tests.py:22
Method__init__
(self, diffeq)
tests/DETEST/run.py:13
Method__init__
(self, func, y0, rtol, atol, min_step=0, max_step=float('inf'), solver="LSODA", **unused_kwargs)
torchdiffeq/_impl/scipy_wrapper.py:9
Method__init__
(self, func, y0, rtol, atol, min_step=0, max_step=float('inf'),
torchdiffeq/_impl/rk_common.py:166
Method__init__
(self, base_func, shapes)
torchdiffeq/_impl/misc.py:149
Method__init__
(self, base_func, mul=1.0)
torchdiffeq/_impl/misc.py:159
Method__init__
(self, base_func)
torchdiffeq/_impl/misc.py:176
Method__init__
(self, func, y0, rtol=1e-3, atol=1e-4, implicit=True, max_iters=_MAX_ITERS, max_order=_MAX_ORDER,
torchdiffeq/_impl/fixed_adams.py:167
Method__init__
(self, dtype, y0, norm, **unused_kwargs)
torchdiffeq/_impl/solvers.py:8
Method__init__
(self, func, y0, step_size=None, grid_constructor=None, interp="linear", perturb=False, **unused_kwargs)
torchdiffeq/_impl/solvers.py:55
Method__init__
(self, radius=0.2, gravity=9.8, adjoint=False)
examples/bouncing_ball.py:15
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
examples/odenet_mnist.py:51
Method__init__
(self, dim_in, dim_out, ksize=3, stride=1, padding=0, dilation=1, groups=1, bias=True, transpose=False)
examples/odenet_mnist.py:78
Method__init__
(self, odefunc)
examples/odenet_mnist.py:118
Method__init__
(self)
examples/odenet_mnist.py:139
Method__init__
(self, momentum=0.99)
examples/odenet_mnist.py:150
Method__init__
(self)
examples/learn_physics.py:15
Method__init__
(self)
examples/learn_physics.py:42
Method__init__
(self)
examples/learn_physics.py:53
Method__init__
(self, aug_dim=2)
examples/learn_physics.py:137
Method__init__
(self)
examples/ode_demo.py:113
Method__init__
(self, momentum=0.99)
examples/ode_demo.py:134
Method__init__
(self, latent_dim=4, nhidden=20)
examples/latent_ode.py:110
Method__init__
(self, latent_dim=4, obs_dim=2, nhidden=25, nbatch=1)
examples/latent_ode.py:130
Method__init__
(self, momentum=0.99)
examples/latent_ode.py:165
Method__init__
(self, in_out_dim, hidden_dim, width)
examples/cnf.py:83
Method__init__
(self, momentum=0.99)
examples/cnf.py:120
Function_adjoint_norm
(tensor_tuple)
torchdiffeq/_impl/adjoint.py:283
Method_advance
Interpolate through the next time point, integrating as necessary.
torchdiffeq/_impl/rk_common.py:243
Method_advance_until_event
Returns t, state(t) such that event_fn(t, state(t)) == 0.
torchdiffeq/_impl/rk_common.py:252
Function_assert_one_dimensional
(name, t)
torchdiffeq/_impl/misc.py:102
Method_before_integrate
(self, t)
torchdiffeq/_impl/rk_common.py:213
Function_decreasing
(t)
torchdiffeq/_impl/misc.py:98
Method_grid_constructor
(func, y0, t)
torchdiffeq/_impl/solvers.py:87
Function_linf_norm
(tensor)
torchdiffeq/_impl/misc.py:18
Function_norm
(tensor)
torchdiffeq/_impl/misc.py:251
Method_step_func
(self, func, t0, dt, t1, y0)
torchdiffeq/_impl/rk_common.py:415
Method_step_func
(self, func, t0, dt, t1, y0)
torchdiffeq/_impl/rk_common.py:488
Method_step_func
(self, func, t0, dt, t1, y0)
torchdiffeq/_impl/fixed_grid.py:9
Method_step_func
(self, func, t0, dt, t1, y0)
torchdiffeq/_impl/fixed_grid.py:17
Method_step_func
(self, func, t0, dt, t1, y0)
torchdiffeq/_impl/fixed_grid.py:27
Method_step_func
(self, func, t0, dt, t1, y0)
torchdiffeq/_impl/fixed_grid.py:35
Method_step_func
(self, func, t0, dt, t1, y0)
torchdiffeq/_impl/fixed_grid.py:51
Method_step_func
(self, func, t0, dt, t1, y0)
torchdiffeq/_impl/fixed_adams.py:195
Function_to_tensor
(x)
tests/DETEST/detest.py:331
Function_zero_norm
(tensor)
torchdiffeq/_impl/misc.py:26
Methodactual_norm
(tensor_tuple)
tests/norm_tests.py:132
Methodadjoint_norm
(tensor_tuple)
tests/norm_tests.py:201
Functionadjoint_seminorm
(tensor_tuple)
torchdiffeq/_impl/adjoint.py:267
Methodaugmented_dynamics
(t, y_aug)
torchdiffeq/_impl/adjoint.py:72
Methodbackward
(ctx, grad_scratch)
torchdiffeq/_impl/rk_common.py:39
Methodbackward
(ctx, *grad_y)
torchdiffeq/_impl/adjoint.py:36
Methodbackward
(ctx, grad_t, grad_state)
torchdiffeq/_impl/odeint.py:208
Functioncallback
(t0, y0, dt, _callback=callback)
torchdiffeq/_impl/misc.py:326
Methodcallback_accept_step_adjoint
(t0, y0, dt)
tests/odeint_tests.py:350
Methodcallback_reject_step_adjoint
(t0, y0, dt)
tests/odeint_tests.py:354
Methodcallback_step_adjoint
(t0, y0, dt)
tests/odeint_tests.py:346
Functioncombined_event_fn
(t, y)
torchdiffeq/_impl/event_handling.py:31
Functiondefault_adjoint_norm
(tensor_tuple)
torchdiffeq/_impl/adjoint.py:247
Functiondense_output_fn
(t_eval)
torchdiffeq/_impl/odeint.py:150
Functiondiffeq
(t, y)
tests/DETEST/detest.py:47
Methodevent_fn
(t, y)
tests/event_tests.py:38
Methodevent_fn
(self, t, state)
examples/bouncing_ball.py:31
← previousnext →101–200 of 279, ranked by callers