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 46e3e72 commit d00f9e2Copy full SHA for d00f9e2
pytensor/link/pytorch/dispatch/elemwise.py
@@ -1,4 +1,5 @@
1
import importlib
2
+
3
import torch
4
5
from pytensor.link.pytorch.dispatch.basic import pytorch_funcify
pytensor/link/pytorch/dispatch/scalar.py
@@ -8,8 +8,8 @@
8
Cast,
9
ScalarOp,
10
)
11
-from pytensor.scalar.math import Softplus
12
from pytensor.scalar.loop import ScalarLoop
+from pytensor.scalar.math import Softplus
13
14
15
@pytorch_funcify.register(ScalarOp)
@@ -60,6 +60,7 @@ def cast(x):
60
61
return cast
62
63
64
@pytorch_funcify.register(Softplus)
65
def pytorch_funcify_Softplus(op, node, **kwargs):
66
return torch.nn.Softplus()
0 commit comments