einsum-计算不同维度的矩阵乘法怎么实现
收藏
我需要实现PyTorch中einsum函数的一个功能,pytorch代码如下:
torch.einsum('bkhw,bckhw->bchw', [W, X])
四维矩阵与五维矩阵相乘,两个矩阵第0维和后两个维度相同,如代码所示,其余维度k维度中元素对应相乘后相加,将此标量放在c维度中,最后输出维度为bchw。
请问用paddle如何实现,求各路大神帮忙想想办法?
0
收藏
请登录后评论
python版,没有并行优化,后续会改进成算子的
啊我错了,还不能反向传播