张量流中的张量乘法(具有不确定的轴数)

时间:2016-11-07 10:29:38

标签: python tensorflow linear-algebra numpy-einsum

我的张量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的等级未知时我该怎么办?是否存在某种有效的工作方式?

0 个答案:

没有答案