Skip to content

Commit 1ffd7c6

Browse files
author
Ian Schweer
committed
Fix comment
1 parent 1865de9 commit 1ffd7c6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytensor/link/pytorch/dispatch/elemwise.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,8 +186,8 @@ def softmax_grad(dy, sm):
186186
def elemwise_scalar_loop(base_fn, op, node, **kwargs):
187187
"""
188188
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
189+
to not work, but https://github.com/pymc-devs/pytensor/issues/1031,
190+
vmap won't allow it. Instead, we can ravel all the inputs, broadcasted
191191
according to torch
192192
"""
193193

0 commit comments

Comments
 (0)