张量流中的张量乘法（轴数不确定）

Question

我有一个张量a，轴数未知（但至少有一个）和一个正方形矩阵M，这样a.get_shape()[0] == M.get_shape()[0]==M.get_shape()[1]。在

我想做的是

R = tf.einsum("i...,ij->j...",a,M)

但是与numpy einsum不同，tensorfloweinsum最近在tensorflow的版本11中实现了，它确实支持使用...，因此，如果我知道{}的秩为4，那么我就可以解决我的问题了

^{pr2}$

但是当a的秩未知时，我该怎么办呢？有没有什么高效的工作？在