Skip to content

Commit 5417a3c

Browse files
committed
Fix bug in ZeroGrad
1 parent 0ea61bc commit 5417a3c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/gradient.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2302,7 +2302,7 @@ def _is_zero(x):
23022302

23032303
class ZeroGrad(ViewOp):
23042304
def grad(self, args, g_outs):
2305-
return [g_out.zeros_like(g_out) for g_out in g_outs]
2305+
return [g_out.zeros_like() for g_out in g_outs]
23062306

23072307
def R_op(self, inputs, eval_points):
23082308
if eval_points[0] is None:

0 commit comments

Comments
 (0)