code: def CONTINUOUS_FUN(p, x):
      return jnp.expand_dims(jnp.sum(p * (x**2)), -1)
  
imports: import jax
  from jax import grad, jit, make_jaxpr
  from jax import numpy
  from plum.function import NotFoundLookupError
  from pytest import mark, raises
  from jaxdf import *
  from jaxdf.operators.dummy import dummy, yummy
  from jaxdf.signatures import SignatureError
label_desc: Use image node_id 1 for calling a global function with example usage: CONTINUOUS_FUN(p, x) and returns: jnp