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 9c119f7 commit 7dd9edbCopy full SHA for 7dd9edb
pytensor/link/pytorch/dispatch/elemwise.py
@@ -32,7 +32,6 @@ def elemwise_fn(*inputs):
32
for _ in range(broadcast_inputs[0].dim()):
33
ufunc = torch.vmap(ufunc)
34
return ufunc(*broadcast_inputs)
35
- return base_fn(*inputs)
36
37
return elemwise_fn
38
0 commit comments