Skip to content

Commit eb7c3b6

Browse files
committed
Use set_subtensor instead of inc_subtensor in Ordered transform
1 parent 73dee50 commit eb7c3b6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pymc/distributions/transforms.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -96,14 +96,14 @@ def __init__(self, ndim_supp=None):
9696

9797
def backward(self, value, *inputs):
9898
x = pt.zeros(value.shape)
99-
x = pt.inc_subtensor(x[..., 0], value[..., 0])
100-
x = pt.inc_subtensor(x[..., 1:], pt.exp(value[..., 1:]))
99+
x = pt.set_subtensor(x[..., 0], value[..., 0])
100+
x = pt.set_subtensor(x[..., 1:], pt.exp(value[..., 1:]))
101101
return pt.cumsum(x, axis=-1)
102102

103103
def forward(self, value, *inputs):
104104
y = pt.zeros(value.shape)
105-
y = pt.inc_subtensor(y[..., 0], value[..., 0])
106-
y = pt.inc_subtensor(y[..., 1:], pt.log(value[..., 1:] - value[..., :-1]))
105+
y = pt.set_subtensor(y[..., 0], value[..., 0])
106+
y = pt.set_subtensor(y[..., 1:], pt.log(value[..., 1:] - value[..., :-1]))
107107
return y
108108

109109
def log_jac_det(self, value, *inputs):

0 commit comments

Comments
 (0)