tensorflow 一个矩阵与多个矩阵相乘时的计算方法(二维和三维张量相乘为例)

当tensor1 的 shape 为[k, m, n], tensor2 的 shape 为 [n, p]时,html 要将tensor1的后两维构成的k个矩阵与tensor2中的矩阵作矩阵乘法获得 shape 为[k, m, p]的向量dom   解决办法:code 1,咱们知道TensorFlow的matmul已经支持了batch,即:orm A = tf.Variable(tf.random
相关文章
相关标签/搜索