Numpy TensorFlow:计算求和块矩阵乘法的有效方法

Numpy TensorFlow:计算求和块矩阵乘法的有效方法,numpy,tensorflow,Numpy,Tensorflow,我有以下等式: 其中M是[Dx3]矩阵,V是[DxD]矩阵。每一个都在一个更大的[3Kx3K]矩阵中形成一个[3x3]块,由i,j索引。现在,我想知道是否有人在TensorFlow中遇到过这种形式的求和-我仍然习惯于API结构

我有以下等式:

其中M是[Dx3]矩阵,V是[DxD]矩阵。每一个都在一个更大的[3Kx3K]矩阵中形成一个[3x3]块,由i,j索引。现在,我想知道是否有人在TensorFlow中遇到过这种形式的求和-我仍然习惯于API结构