API tools faq
paste
Advertisement
Guest User

Untitled

a guest
Oct 20th, 2020
64
0
Never
Add comment
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 31.43 KB | None | 0 0
raw download clone embed print report
  1. ```
  2. ---------------------------------------------------------------------------
  3. FilteredStackTrace Traceback (most recent call last)
  4. <ipython-input-9-10a631dc4096> in <module>
  5. 15 dlogp_logp_fn = jax.value_and_grad(logp_fn_jax, argnums=range(len(init_state)))
  6. ---> 16 jax.vmap(dlogp_logp_fn)(*init_state_batched)
  7.  
  8. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  9. 165 def jax_func(*inputs):
  10. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  11. 167 return return_func(*func_args)
  12.  
  13. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  14. 165 def jax_func(*inputs):
  15. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  16. 167 return return_func(*func_args)
  17.  
  18. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  19. 165 def jax_func(*inputs):
  20. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  21. 167 return return_func(*func_args)
  22.  
  23. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  24. 165 def jax_func(*inputs):
  25. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  26. 167 return return_func(*func_args)
  27.  
  28. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  29. 165 def jax_func(*inputs):
  30. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  31. 167 return return_func(*func_args)
  32.  
  33. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  34. 165 def jax_func(*inputs):
  35. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  36. 167 return return_func(*func_args)
  37.  
  38. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  39. 165 def jax_func(*inputs):
  40. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  41. 167 return return_func(*func_args)
  42.  
  43. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  44. 165 def jax_func(*inputs):
  45. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  46. 167 return return_func(*func_args)
  47.  
  48. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  49. 165 def jax_func(*inputs):
  50. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  51. 167 return return_func(*func_args)
  52.  
  53. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  54. 165 def jax_func(*inputs):
  55. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  56. 167 return return_func(*func_args)
  57.  
  58. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  59. 165 def jax_func(*inputs):
  60. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  61. 167 return return_func(*func_args)
  62.  
  63. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  64. 165 def jax_func(*inputs):
  65. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  66. 167 return return_func(*func_args)
  67.  
  68. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  69. 165 def jax_func(*inputs):
  70. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  71. 167 return return_func(*func_args)
  72.  
  73. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  74. 165 def jax_func(*inputs):
  75. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  76. 167 return return_func(*func_args)
  77.  
  78. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  79. 165 def jax_func(*inputs):
  80. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  81. 167 return return_func(*func_args)
  82.  
  83. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  84. 165 def jax_func(*inputs):
  85. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  86. 167 return return_func(*func_args)
  87.  
  88. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  89. 165 def jax_func(*inputs):
  90. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  91. 167 return return_func(*func_args)
  92.  
  93. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  94. 165 def jax_func(*inputs):
  95. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  96. 167 return return_func(*func_args)
  97.  
  98. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  99. 165 def jax_func(*inputs):
  100. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  101. 167 return return_func(*func_args)
  102.  
  103. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  104. 165 def jax_func(*inputs):
  105. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  106. 167 return return_func(*func_args)
  107.  
  108. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  109. 165 def jax_func(*inputs):
  110. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  111. 167 return return_func(*func_args)
  112.  
  113. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  114. 165 def jax_func(*inputs):
  115. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  116. 167 return return_func(*func_args)
  117.  
  118. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  119. 165 def jax_func(*inputs):
  120. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  121. 167 return return_func(*func_args)
  122.  
  123. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  124. 165 def jax_func(*inputs):
  125. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  126. 167 return return_func(*func_args)
  127.  
  128. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  129. 165 def jax_func(*inputs):
  130. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  131. 167 return return_func(*func_args)
  132.  
  133. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  134. 165 def jax_func(*inputs):
  135. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  136. 167 return return_func(*func_args)
  137.  
  138. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  139. 165 def jax_func(*inputs):
  140. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  141. 167 return return_func(*func_args)
  142.  
  143. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  144. 165 def jax_func(*inputs):
  145. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  146. 167 return return_func(*func_args)
  147.  
  148. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  149. 165 def jax_func(*inputs):
  150. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  151. 167 return return_func(*func_args)
  152.  
  153. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  154. 165 def jax_func(*inputs):
  155. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  156. 167 return return_func(*func_args)
  157.  
  158. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  159. 165 def jax_func(*inputs):
  160. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  161. 167 return return_func(*func_args)
  162.  
  163. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  164. 165 def jax_func(*inputs):
  165. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  166. 167 return return_func(*func_args)
  167.  
  168. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  169. 165 def jax_func(*inputs):
  170. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  171. 167 return return_func(*func_args)
  172.  
  173. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  174. 165 def jax_func(*inputs):
  175. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  176. 167 return return_func(*func_args)
  177.  
  178. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  179. 165 def jax_func(*inputs):
  180. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  181. 167 return return_func(*func_args)
  182.  
  183. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  184. 165 def jax_func(*inputs):
  185. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  186. 167 return return_func(*func_args)
  187.  
  188. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  189. 165 def jax_func(*inputs):
  190. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  191. 167 return return_func(*func_args)
  192.  
  193. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  194. 165 def jax_func(*inputs):
  195. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  196. 167 return return_func(*func_args)
  197.  
  198. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  199. 165 def jax_func(*inputs):
  200. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  201. 167 return return_func(*func_args)
  202.  
  203. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  204. 165 def jax_func(*inputs):
  205. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  206. 167 return return_func(*func_args)
  207.  
  208. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  209. 165 def jax_func(*inputs):
  210. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  211. 167 return return_func(*func_args)
  212.  
  213. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  214. 165 def jax_func(*inputs):
  215. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  216. 167 return return_func(*func_args)
  217.  
  218. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  219. 165 def jax_func(*inputs):
  220. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  221. 167 return return_func(*func_args)
  222.  
  223. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  224. 165 def jax_func(*inputs):
  225. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  226. 167 return return_func(*func_args)
  227.  
  228. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  229. 165 def jax_func(*inputs):
  230. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  231. 167 return return_func(*func_args)
  232.  
  233. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  234. 165 def jax_func(*inputs):
  235. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  236. 167 return return_func(*func_args)
  237.  
  238. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  239. 165 def jax_func(*inputs):
  240. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  241. 167 return return_func(*func_args)
  242.  
  243. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  244. 165 def jax_func(*inputs):
  245. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  246. 167 return return_func(*func_args)
  247.  
  248. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  249. 165 def jax_func(*inputs):
  250. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  251. 167 return return_func(*func_args)
  252.  
  253. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  254. 165 def jax_func(*inputs):
  255. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  256. 167 return return_func(*func_args)
  257.  
  258. FilteredStackTrace: TypeError: 'list' object is not callable
  259.  
  260. The stack trace above excludes JAX-internal frames.
  261. The following is the original exception that occurred, unmodified.
  262.  
  263. --------------------
  264.  
  265. The above exception was the direct cause of the following exception:
  266.  
  267. TypeError Traceback (most recent call last)
  268. <ipython-input-9-10a631dc4096> in <module>
  269. 14
  270. 15 dlogp_logp_fn = jax.value_and_grad(logp_fn_jax, argnums=range(len(init_state)))
  271. ---> 16 jax.vmap(dlogp_logp_fn)(*init_state_batched)
  272.  
  273. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/traceback_util.py in reraise_with_filtered_traceback(*args, **kwargs)
  274. 137 def reraise_with_filtered_traceback(*args, **kwargs):
  275. 138 try:
  276. --> 139 return fun(*args, **kwargs)
  277. 140 except Exception as e:
  278. 141 if not is_under_reraiser(e):
  279.  
  280. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/api.py in batched_fun(*args)
  281. 1225 in_axes_flat = flatten_axes("vmap in_axes", in_tree, in_axes)
  282. 1226 _ = _mapped_axis_size(in_tree, args_flat, in_axes_flat, "vmap")
  283. -> 1227 out_flat = batching.batch(flat_fun, args_flat, in_axes_flat,
  284. 1228 lambda: flatten_axes("vmap out_axes", out_tree(),
  285. 1229 out_axes),
  286.  
  287. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/interpreters/batching.py in batch(fun, in_vals, in_dims, out_dim_dests, axis_name)
  288. 34 # executes a batched version of `fun` following out_dim_dests
  289. 35 batched_fun = batch_fun(fun, in_dims, out_dim_dests, axis_name=axis_name)
  290. ---> 36 return batched_fun.call_wrapped(*in_vals)
  291. 37
  292. 38 @lu.transformation_with_aux
  293.  
  294. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/linear_util.py in call_wrapped(self, *args, **kwargs)
  295. 154
  296. 155 try:
  297. --> 156 ans = self.f(*args, **dict(self.params, **kwargs))
  298. 157 except:
  299. 158 # Some transformations yield from inside context managers, so we have to
  300.  
  301. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/traceback_util.py in reraise_with_filtered_traceback(*args, **kwargs)
  302. 137 def reraise_with_filtered_traceback(*args, **kwargs):
  303. 138 try:
  304. --> 139 return fun(*args, **kwargs)
  305. 140 except Exception as e:
  306. 141 if not is_under_reraiser(e):
  307.  
  308. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/api.py in value_and_grad_f(*args, **kwargs)
  309. 808 tree_map(partial(_check_input_dtype_grad, holomorphic, allow_int), dyn_args)
  310. 809 if not has_aux:
  311. --> 810 ans, vjp_py = _vjp(f_partial, *dyn_args)
  312. 811 else:
  313. 812 ans, vjp_py, aux = _vjp(f_partial, *dyn_args, has_aux=True)
  314.  
  315. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/api.py in _vjp(fun, has_aux, *primals)
  316. 1839 if not has_aux:
  317. 1840 flat_fun, out_tree = flatten_fun_nokwargs(fun, in_tree)
  318. -> 1841 out_primal, out_vjp = ad.vjp(flat_fun, primals_flat)
  319. 1842 out_tree = out_tree()
  320. 1843 else:
  321.  
  322. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/interpreters/ad.py in vjp(traceable, primals, has_aux)
  323. 111 def vjp(traceable, primals, has_aux=False):
  324. 112 if not has_aux:
  325. --> 113 out_primals, pvals, jaxpr, consts = linearize(traceable, *primals)
  326. 114 else:
  327. 115 out_primals, pvals, jaxpr, consts, aux = linearize(traceable, *primals, has_aux=True)
  328.  
  329. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/interpreters/ad.py in linearize(traceable, *primals, **kwargs)
  330. 98 _, in_tree = tree_flatten(((primals, primals), {}))
  331. 99 jvpfun_flat, out_tree = flatten_fun(jvpfun, in_tree)
  332. --> 100 jaxpr, out_pvals, consts = pe.trace_to_jaxpr(jvpfun_flat, in_pvals)
  333. 101 out_primals_pvals, out_tangents_pvals = tree_unflatten(out_tree(), out_pvals)
  334. 102 assert all(out_primal_pval.is_known() for out_primal_pval in out_primals_pvals)
  335.  
  336. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/interpreters/partial_eval.py in trace_to_jaxpr(fun, pvals, instantiate, stage_out, bottom, trace_type)
  337. 1115 with core.new_main(trace_type, bottom=bottom) as main:
  338. 1116 fun = trace_to_subjaxpr(fun, main, instantiate)
  339. -> 1117 jaxpr, (out_pvals, consts, env) = fun.call_wrapped(pvals)
  340. 1118 assert not env
  341. 1119 del main
  342.  
  343. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/jax/linear_util.py in call_wrapped(self, *args, **kwargs)
  344. 154
  345. 155 try:
  346. --> 156 ans = self.f(*args, **dict(self.params, **kwargs))
  347. 157 except:
  348. 158 # Some transformations yield from inside context managers, so we have to
  349.  
  350. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  351. 164
  352. 165 def jax_func(*inputs):
  353. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  354. 167 return return_func(*func_args)
  355. 168
  356.  
  357. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  358. 164
  359. 165 def jax_func(*inputs):
  360. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  361. 167 return return_func(*func_args)
  362. 168
  363.  
  364. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  365. 164
  366. 165 def jax_func(*inputs):
  367. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  368. 167 return return_func(*func_args)
  369. 168
  370.  
  371. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  372. 164
  373. 165 def jax_func(*inputs):
  374. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  375. 167 return return_func(*func_args)
  376. 168
  377.  
  378. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  379. 164
  380. 165 def jax_func(*inputs):
  381. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  382. 167 return return_func(*func_args)
  383. 168
  384.  
  385. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  386. 164
  387. 165 def jax_func(*inputs):
  388. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  389. 167 return return_func(*func_args)
  390. 168
  391.  
  392. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  393. 164
  394. 165 def jax_func(*inputs):
  395. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  396. 167 return return_func(*func_args)
  397. 168
  398.  
  399. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  400. 164
  401. 165 def jax_func(*inputs):
  402. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  403. 167 return return_func(*func_args)
  404. 168
  405.  
  406. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  407. 164
  408. 165 def jax_func(*inputs):
  409. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  410. 167 return return_func(*func_args)
  411. 168
  412.  
  413. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  414. 164
  415. 165 def jax_func(*inputs):
  416. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  417. 167 return return_func(*func_args)
  418. 168
  419.  
  420. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  421. 164
  422. 165 def jax_func(*inputs):
  423. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  424. 167 return return_func(*func_args)
  425. 168
  426.  
  427. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  428. 164
  429. 165 def jax_func(*inputs):
  430. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  431. 167 return return_func(*func_args)
  432. 168
  433.  
  434. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  435. 164
  436. 165 def jax_func(*inputs):
  437. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  438. 167 return return_func(*func_args)
  439. 168
  440.  
  441. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  442. 164
  443. 165 def jax_func(*inputs):
  444. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  445. 167 return return_func(*func_args)
  446. 168
  447.  
  448. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  449. 164
  450. 165 def jax_func(*inputs):
  451. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  452. 167 return return_func(*func_args)
  453. 168
  454.  
  455. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  456. 164
  457. 165 def jax_func(*inputs):
  458. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  459. 167 return return_func(*func_args)
  460. 168
  461.  
  462. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  463. 164
  464. 165 def jax_func(*inputs):
  465. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  466. 167 return return_func(*func_args)
  467. 168
  468.  
  469. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  470. 164
  471. 165 def jax_func(*inputs):
  472. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  473. 167 return return_func(*func_args)
  474. 168
  475.  
  476. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  477. 164
  478. 165 def jax_func(*inputs):
  479. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  480. 167 return return_func(*func_args)
  481. 168
  482.  
  483. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  484. 164
  485. 165 def jax_func(*inputs):
  486. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  487. 167 return return_func(*func_args)
  488. 168
  489.  
  490. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  491. 164
  492. 165 def jax_func(*inputs):
  493. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  494. 167 return return_func(*func_args)
  495. 168
  496.  
  497. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  498. 164
  499. 165 def jax_func(*inputs):
  500. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  501. 167 return return_func(*func_args)
  502. 168
  503.  
  504. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  505. 164
  506. 165 def jax_func(*inputs):
  507. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  508. 167 return return_func(*func_args)
  509. 168
  510.  
  511. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  512. 164
  513. 165 def jax_func(*inputs):
  514. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  515. 167 return return_func(*func_args)
  516. 168
  517.  
  518. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  519. 164
  520. 165 def jax_func(*inputs):
  521. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  522. 167 return return_func(*func_args)
  523. 168
  524.  
  525. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  526. 164
  527. 165 def jax_func(*inputs):
  528. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  529. 167 return return_func(*func_args)
  530. 168
  531.  
  532. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  533. 164
  534. 165 def jax_func(*inputs):
  535. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  536. 167 return return_func(*func_args)
  537. 168
  538.  
  539. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  540. 164
  541. 165 def jax_func(*inputs):
  542. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  543. 167 return return_func(*func_args)
  544. 168
  545.  
  546. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  547. 164
  548. 165 def jax_func(*inputs):
  549. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  550. 167 return return_func(*func_args)
  551. 168
  552.  
  553. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  554. 164
  555. 165 def jax_func(*inputs):
  556. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  557. 167 return return_func(*func_args)
  558. 168
  559.  
  560. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  561. 164
  562. 165 def jax_func(*inputs):
  563. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  564. 167 return return_func(*func_args)
  565. 168
  566.  
  567. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  568. 164
  569. 165 def jax_func(*inputs):
  570. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  571. 167 return return_func(*func_args)
  572. 168
  573.  
  574. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  575. 164
  576. 165 def jax_func(*inputs):
  577. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  578. 167 return return_func(*func_args)
  579. 168
  580.  
  581. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  582. 164
  583. 165 def jax_func(*inputs):
  584. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  585. 167 return return_func(*func_args)
  586. 168
  587.  
  588. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  589. 164
  590. 165 def jax_func(*inputs):
  591. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  592. 167 return return_func(*func_args)
  593. 168
  594.  
  595. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  596. 164
  597. 165 def jax_func(*inputs):
  598. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  599. 167 return return_func(*func_args)
  600. 168
  601.  
  602. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  603. 164
  604. 165 def jax_func(*inputs):
  605. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  606. 167 return return_func(*func_args)
  607. 168
  608.  
  609. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  610. 164
  611. 165 def jax_func(*inputs):
  612. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  613. 167 return return_func(*func_args)
  614. 168
  615.  
  616. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  617. 164
  618. 165 def jax_func(*inputs):
  619. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  620. 167 return return_func(*func_args)
  621. 168
  622.  
  623. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  624. 164
  625. 165 def jax_func(*inputs):
  626. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  627. 167 return return_func(*func_args)
  628. 168
  629.  
  630. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  631. 164
  632. 165 def jax_func(*inputs):
  633. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  634. 167 return return_func(*func_args)
  635. 168
  636.  
  637. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  638. 164
  639. 165 def jax_func(*inputs):
  640. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  641. 167 return return_func(*func_args)
  642. 168
  643.  
  644. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  645. 164
  646. 165 def jax_func(*inputs):
  647. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  648. 167 return return_func(*func_args)
  649. 168
  650.  
  651. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  652. 164
  653. 165 def jax_func(*inputs):
  654. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  655. 167 return return_func(*func_args)
  656. 168
  657.  
  658. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  659. 164
  660. 165 def jax_func(*inputs):
  661. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  662. 167 return return_func(*func_args)
  663. 168
  664.  
  665. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  666. 164
  667. 165 def jax_func(*inputs):
  668. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  669. 167 return return_func(*func_args)
  670. 168
  671.  
  672. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  673. 164
  674. 165 def jax_func(*inputs):
  675. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  676. 167 return return_func(*func_args)
  677. 168
  678.  
  679. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  680. 164
  681. 165 def jax_func(*inputs):
  682. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  683. 167 return return_func(*func_args)
  684. 168
  685.  
  686. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in jax_func(*inputs)
  687. 164
  688. 165 def jax_func(*inputs):
  689. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  690. 167 return return_func(*func_args)
  691. 168
  692.  
  693. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/theano/sandbox/jaxify.py in <listcomp>(.0)
  694. 164
  695. 165 def jax_func(*inputs):
  696. --> 166 func_args = [fn(*inputs) for fn in input_funcs]
  697. 167 return return_func(*func_args)
  698. 168
  699.  
  700. TypeError: 'list' object is not callable
  701. ```
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
Public Pastes
Advertisement
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy.  OK, I Understand
Not a member of Pastebin yet?
Sign Up, it unlocks many cool features!