Python to_scalar Examples

Example #1

File: numpy_utils.py Project: agibsonccc/autograd

 def d_scalar_fun(*args):
     return to_scalar(grad(scalar_fun, argnum=i)(*args))

Example #2

File: numpy_utils.py Project: agibsonccc/autograd

 def scalar_fun(*new_args):
     full_args = list(args)
     for i, argnum in enumerate(argnums):
         full_args[argnum] = new_args[i]
     return to_scalar(fun(*full_args, **kwargs))

Example #3

 def scalar_fun(*new_args):
     full_args = list(args)
     for i, argnum in enumerate(argnums):
         full_args[argnum] = new_args[i]
     return to_scalar(fun(*full_args, **kwargs))

Example #4

 def d_scalar_fun(*args):
     return to_scalar(grad(scalar_fun, argnum=i)(*args))

Example #5

File: numpy_utils.py Project: vegaandagev/autograd-forward

 def scalar_args_fun(*new_args):
     full_args = list(args)
     for i, argnum in enumerate(argnums):
         wrt_flat, unflatten = flatten(wrt_args[i])
         full_args[argnum] = unflatten(wrt_flat + new_args[i] * rand_vecs[i])
     return to_scalar(fun(*full_args, **kwargs))