We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5a1d82b commit 9c119f7Copy full SHA for 9c119f7
pytensor/link/pytorch/dispatch/elemwise.py
@@ -172,8 +172,8 @@ def softmax_grad(dy, sm):
172
def elemwise_scalar_loop(base_fn, op, node, **kwargs):
173
"""
174
ScalarLoop + Elemwise is too common
175
- to not work, but @1031, vmap won't allow it.
176
- Instead, we can ravel all the inputs, broadcasted
+ to not work, but https://github.com/pymc-devs/pytensor/issues/1031,
+ vmap won't allow it. Instead, we can ravel all the inputs, broadcasted
177
according to torch
178
179
0 commit comments