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 1ffd7c6 commit fd2f192Copy full SHA for fd2f192
pytensor/link/pytorch/dispatch/elemwise.py
@@ -46,7 +46,6 @@ def elemwise_fn(*inputs):
46
for _ in range(broadcast_inputs[0].dim()):
47
ufunc = torch.vmap(ufunc)
48
return ufunc(*broadcast_inputs)
49
- return base_fn(*inputs)
50
51
return elemwise_fn
52
0 commit comments