MCPcopy
hub / github.com/LuChengTHU/dpm-solver / batch_mul

Function batch_mul

examples/score_sde_jax/utils.py:41–42  ·  view source on GitHub ↗
(a, b)

Source from the content-addressed store, hash-verified

39
40
41def batch_mul(a, b):
42 return jax.vmap(lambda a, b: a * b)(a, b)
43
44
45def load_training_state(filepath, state):

Callers 15

sdeMethod · 0.90
discretizeMethod · 0.90
sdeMethod · 0.90
marginal_probMethod · 0.90
discretizeMethod · 0.90
sdeMethod · 0.90
marginal_probMethod · 0.90
inpaint_update_fnFunction · 0.90
colorization_update_fnFunction · 0.90
loss_fnFunction · 0.90
update_fnMethod · 0.90
update_fnMethod · 0.90

Calls

no outgoing calls

Tested by

no test coverage detected