我想创建一个函数fn，该函数收缩一组3索引张量的索引。我希望张量N为fn的参数。例如，假设N = 3。然后，fn（N = 3）应该返回：
tf.einsum("sab,zbc,xcd->szxad", A1, A2, A3)
对于N = 4：
tf.einsum("sab,zbc,xcd,tde->szxtae", A1, A2, A3, A4)
依此类推...
我认为tf.while_loop可能是可行的方法，但是由于tf.einsum中指定的索引数随N改变，所以我不知道如何对普通N执行此操作。
工作是什么功能？