分类：matmul

2021
07-01

pytorch中的matmul与mm,bmm区别说明

pytorch中matmul和mm和bmm区别matmulmmbmm结论先看下官网上对这三个函数的介绍。matmulmmbmm顾名思义,就是两个batch矩阵乘法.结论从官方文档可以看出1、mm只能进行矩阵乘法,也就是输入的两个tensor维度只能是(n×m)(n\timesm)(n×m)和(m×p)(m\timesp)(m×p)2、bmm是两个三维张量相乘,两个输入tensor维度是(b×n×m)(b\timesn\timesm)(b×n×m)和(b×m×p)(b\timesm\timesp)(b×m×p),第一维b代... 继续阅读 >

2020
10-08

浅谈keras中的batch_dot,dot方法和TensorFlow的matmul

概述在使用keras中的keras.backend.batch_dot和tf.matmul实现功能其实是一样的智能矩阵乘法，比如A,B,C,D,E,F,G,H，I,J,K,L都是二维矩阵，中间点表示矩阵乘法，AG表示矩阵A和G矩阵乘法（A的列维度等于G行维度），WX=Zimportkeras.backendasKimporttensorflowastfimportnumpyasnpw=K.variable(np.random.randint(10,size=(10,12,4,5)))k=K.variable(np.random.randint(10,size=(10,12,5,8)))z=K.batch_dot(w,k)... 继续阅读 >