使用 Tensorflow ODE 求解器获取梯度时出现 ZeroDivisionError 错误？

如何解决使用 Tensorflow ODE 求解器获取梯度时出现 ZeroDivisionError 错误？

我正在尝试使用 TensorFlow 实现一个机械模型，该模型将用作 GAN 的一部分，基于本文所示的方法：https://arxiv.org/abs/2009.08267。我正在使用 tensorflow 2.5.0 和 tensorflow-probability 0.13.0。

机械模型使用 TF Dormand-Prince 求解器来求解一组微分方程，这些方程为心血管系统的不同区域生成压力波形。我想获得关于用于训练 GAN 生成器的机械模型参数的波形梯度。

我的几个微分方程包含一个随时间变化的变量（分段但连续，没有“尖角”），并且从参数的子集计算到机械模型。如果我将此变量设置为常数，我可以获得波形的梯度和模型参数。然而，如果我保持这个变量随时间变化，那么当我尝试计算梯度时我会得到一个 ZeroDivisionError。

知道为什么会出现这个错误吗？我在下面包含了一个堆栈跟踪。

非常感谢您的帮助！

<main>
  <h2>Please Select Nominee!</h2>
  <form  name="nominee" onsubmit="VoteSubmit()">
    <label>
      <input type="radio" 
         name="nominee" id="ges" value="Gibson ES-335">
    </label> Gibson ES-335
    <div id="orderTotal1">
      <p id="p1"></p>
      <script>document.write(getPollCount())</script>
    </div>
    <label>
      <input type="radio" 
             name="nominee" id="glp" value="Gibson Les Paul">
    </label>
      Gibson Les Paul
      <div id="orderTotal2">
        <p id="p2"></p>
        <script>document.write(getPollCount())</script>
      </div>
    <label>
      <input type="radio" 
              name="nominee" id="prs" value="Paul Reed Smith">
   </label> 
     Paul Reed Smith
     <div id="orderTotal3">
       <p id = "p3"></p>
       <script>document.write(getPollCount())</script>
     </div>
     <input type="submit" 
         name="select" value="Select" onsubmit="incrementPollCount()">
  </form>
</main>