我的张量a
具有未知数量的轴(但至少有一个)和方形矩阵M
,使得a.get_shape()[0] == M.get_shape()[0]==M.get_shape()[1]
。
我想做的是
R = tf.einsum("i...,ij->j...",a,M)
但与numpy einsum
不同,最近在张量流版本11中实现的张量流einsum
确实支持使用...
,所以如果我知道例如,我可以解决我的问题a
排名第4,然后我可以做
R = tf.einsum("imkl,ij->jmkl",a,M)
但是当a
的等级未知时我该怎么办?是否存在某种有效的工作方式?