TensorFlow：计算缩减和块矩阵乘法的有效方法

Question

我有以下等式：

其中M是[Dx3]矩阵，V是[DxD]矩阵。这些中的每一个在较大的[3Kx3K]矩阵中形成[3×3]块，由i，j索引。现在，我想知道是否有人在TensorFlow中遇到过这种形式的减少 - 我还是习惯了API结构！