如何在 pm.Deterministic() 中插入 for 循环以在 pymc3 模型中创建数组？

如何解决如何在 pm.Deterministic() 中插入 for 循环以在 pymc3 模型中创建数组？

with pm.Model() as model:
  w = pm.Uniform('w',1)
  c = pm.Uniform('c',5)
  b = 0.5
  s = pm.Deterministic('s',pm.math.exp(-c * (w * d1 + (1-w) * d2)))
  s_A = np.zeros((8)) 
  s_A = pm.Deterministic('s_A',for i in range(8) : s_A[i] = pm.math.sum(s[i][j] for j in (0,1,2,4)))
  s_B = np.zeros((8)) 
  s_B = pm.Deterministic('s_B',for i in range(8) : s_B[i] = pm.math.sum(s[i][j] for j in (3,5,6,7)))
  r = pm.Deterministic('r',b*s_A/(b*s_A + (1-b)*s_B))
  y = pm.Binomial('y',n= 320,p = r,observed = y_observed )
  trace = pm.sample(5000)


az.plot_posterior(trace,hdi_prob = 0.95)

我想用上面的参数创建一个pymc3模型。 d1 和 d2 都是 8*8 矩阵。当我尝试为 s_A 和 s_B 数组赋值时，上面的代码显示了语法错误。谁能告诉我如何在 pymc3 模型中创建 s_A 和 s_B 等变量。