我是Keras和Python的初学者。我需要在Keras中实现几个for循环来为张量分配值，但是据我所知，在tensorflow中无法做到这一点。能否请您帮我如何在keras中执行以下代码？如果要对其向量化，该怎么办？如果它是一个for循环，我可以这样做，但是在这里，我有几个for循环，对于张量的每个位置，我需要考虑另一个张量中的所有值。你能告诉我你的建议吗？我真的需要你的建议。谢谢
a和b是张量C(u)和C(v)是使用C函数产生的数字
def DCT(a, b):
for u in range(8):
for v in range(8):
for x in range(8):
for y in range(8):
b[u,v] = b[u, v] + 0.25 * C(u) * C(v) * a[x, y]* np.cos((2 * x+1) * (u) * np.pi / 16) * np.cos((2 * y+1) * (v) * np.pi / 16)