MCPcopy Create free account
hub / github.com/LeCAR-Lab/model-based-diffusion / demo_vector_field_fn

Method demo_vector_field_fn

mbd/scripts/vis_manim.py:44–50  ·  view source on GitHub ↗
(x)

Source from the content-addressed store, hash-verified

42
43 # Create vector field in the box2
44 def demo_vector_field_fn(x):
45 value = (x[0] / 0.6) ** 2 + (x[1] / 1.5) ** 2 - 1.0
46 vec = 0.2 * np.array([x[0] / 0.6, x[1] / 1.5])
47 if value < 0:
48 return vec
49 else:
50 return -vec
51
52 demo_vector_field = ArrowVectorField(
53 demo_vector_field_fn,

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected