张量流图

Question

我试图使用张量流映射函数，但我被困在一个索引问题。你知道吗

在简单的python中，我尝试执行以下操作操作：-你知道吗

for i in range(1,25):
      u [i] = uold [i] - K * ( uold [i] - uold [i-1] )

在tensorflow中，由于“（uold1[I]-uold1[I-1]），我遇到了一个索引问题。目前，我已将声明写成：

 u = tf.map_fn ( lambda u: uold - K * ( uold - uold ), uold )

在当前方程中，第二项总是零。我不知道如何改变它以获得所需的输出。你知道吗

Answer 1

您可能希望创建一个右移一维的张量（使用tf.pad()），然后计算差值。例如

temp = uold_shifted_to_right - K * (uold - uold_shifted_to_right)

然后从temp中取出第一列（使用tf.slice()）。你知道吗