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 1865de9 commit 1ffd7c6Copy full SHA for 1ffd7c6
pytensor/link/pytorch/dispatch/elemwise.py
@@ -186,8 +186,8 @@ def softmax_grad(dy, sm):
186
def elemwise_scalar_loop(base_fn, op, node, **kwargs):
187
"""
188
ScalarLoop + Elemwise is too common
189
- to not work, but @1031, vmap won't allow it.
190
- 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
191
according to torch
192
193
0 commit comments