jax.vmapfor the JAX backend
for-loops in the Python case
for-loops as well
Elemwiseis equivalent to
ufunc, we need an equivalent
at.vectorizewould construct those
sizeparameter with this and generalize a lot of the
sizeparameter would be used when computed)
signature-like information per
Op-level, it can always be provided to a
Scanare very related
Is this behavior of Aesara variables intended/serves some purpose or is it simply a bug ?
import aesara.tensor as aet import copy x = aet.dscalar("x") z = copy.deepcopy(x) assert x==z # Fails
==to work like it generally should, we would need to implement consistent
==effectively does what
g_1 == g_2a little expensive when they're both large graphs that are very similar except in the "leaf" nodes/inputs
dicts all the time, it would surely have an effect
This seems like a whole other problem in itself.
As of now, I was looking for ways to have a copy of a nested structure (for instance a dict with list of TensorVariables), I wanted to track changes within such structures, by comparing old to new modified structures. But ran into this issue when I deepcopied the nested structure.