コード例 #1
0
 def wrapped(*args, **kwargs):
   return harvest.plant(
       f, tag=RANDOM_VARIABLE)(observations, *args, **kwargs)
コード例 #2
0
 def apply_f(variables, *args, **kwargs):
   return harvest.plant(
       f_, tag=module.VARIABLE)(variables, random.PRNGKey(0), *args,
                                **kwargs)
コード例 #3
0
 def program(key, *args, **kwargs):
   return harvest.plant(
       f,
       tag=RANDOM_VARIABLE,
       blocklist=names)({}, key, *args, **kwargs)