Write wrapper function around aesara.scan
with nice, intuitive signature
#6190
michaelosthege
started this conversation in
Ideas
Replies: 1 comment
-
Could it be a decorator for a function that's executed inside the loop? k = tt.iscalar("k")
A = tt.vector("A")
# Compute A**k
@aet.for_(k)
def loop(result=0):
result += result * A
return result |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
We can just take a bunch of kwargs with informative names and docstrings, and behind the scenes feed things into the less intuitive parts of the
aesara.scan
signature such as theoutput_info
kwarg.@ricardoV94
Beta Was this translation helpful? Give feedback.
All reactions