分类：tf.batch_matmul

2021
08-06

TensorFlow中tf.batch_matmul()的用法

TensorFlow中tf.batch_matmul()用法如果有两个三阶张量，size分别为a.shape=[100,3,4]b.shape=[100,4,5]c=tf.batch_matmul(a,b)则c.shape=[100,3,5]//将每一对3x4的矩阵与4x5的矩阵分别相乘。batch_size不变100为张量的batch_size。剩下的两个维度为数据的维度。不过新版的tensorflow已经移除了上面的函数，使用时换为tf.matmul就可以了。与上面注释的方式是同样的。附：如果是更高维度。例如(a,b,m,n)与... 继续阅读 >

2020
10-08

将tf.batch_matmul替换成tf.matmul的实现

我就废话不多说了，大家还是直接看代码吧~importtensorflowastfh_doc=tf.placeholder(tf.int32,[None,30,512])h_query=tf.placeholder(tf.int32,[None,10,512])temp=tf.matmul(h_doc,h_query,adjoint_b=True)#tf.batch_matmul(h_doc,h_query,adj_y=True)print(temp.get_shape())(?,30,10)补充知识：tf.matmul()和tf.multiply()的区别1.tf.multiply（）两个矩阵中对应元素各自相乘格式:tf.multiply(x,y,nam... 继续阅读 >