我有一个a轴数未知（但至少有一个）的张量和一个方阵M，使得a.get_shape()[0] == M.get_shape()[0]==M.get_shape()[1].
我想做的是
R = tf.einsum("i...,ij->j...",a,M)
但与 numpy 不同的是einsum，einsum最近在 tensorflow 版本 11 中实现的 tensorflow 确实支持 using ...，所以如果我知道例如a排名为 4的问题，那么我可以解决我的问题
R = tf.einsum("imkl,ij->jmkl",a,M)
a但是，当等级未知时我该怎么办？是否有某种有效的解决方法？