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 06d1412 commit a5044fcCopy full SHA for a5044fc
pymc3/model.py
@@ -39,7 +39,6 @@
39
import scipy.sparse as sps
40
41
from aesara.compile.sharedvalue import SharedVariable
42
-from aesara.gradient import grad
43
from aesara.graph.basic import Constant, Variable, graph_inputs
44
from aesara.graph.fg import FunctionGraph
45
from aesara.tensor.random.opt import local_subtensor_rv_lift
@@ -446,7 +445,7 @@ def __init__(
446
445
givens.append((var, shared))
447
448
if compute_grads:
449
- grads = grad(cost, grad_vars, disconnected_inputs="ignore")
+ grads = aesara.grad(cost, grad_vars, disconnected_inputs="ignore")
450
for grad_wrt, var in zip(grads, grad_vars):
451
grad_wrt.name = f"{var.name}_grad"
452
outputs = [cost] + grads
0 commit comments