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
-
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