遍历Tensorflow / Keras中的张量

Question

请帮助我在Tensorflow / Keras中实现等效代码。我想我仍然不知道它是如何工作的...

A = np.zeros((N, N))
for i, j, k in zip(input_sequence, input_sequence[1:], output_sequence[1:]):
    A[i, j] += k

input_sequence，output_sequence是形状为（？，100，1）的张量

基本上，这个想法是使用存储在input_sequence中的索引和存储在output_sequence中的值来设置A的值。

例如：

如果input_sequence = [0，1，2，1，2]和output_sequence = [1.1，2.3，3.3，4.5，10.1]，则得出的矩阵为：

[ [0.0, 2.3, 0.0],
  [0.0, 0.0, 13.4],
  [0.0, 4.5, 0.0] ]

请帮助！