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 af92a72 commit cc2bfb4Copy full SHA for cc2bfb4
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