TENSorflow中矩阵和矩阵的有效元素乘法

时间:2017-01-17 03:14:48

标签: tensorflow

最有效的乘法方式是什么: 矩阵A(形状:4m * n,形状为A1:m * n)

A1
A2
A3
A4

由矩阵(形状:4m * n,形状为B1:m * n):

B1
B2
B3
B4

获得一个新矩阵(形状:4m * m):

A1B1
A2B2
A3B3
A4B4

1 个答案:

答案 0 :(得分:0)

这就是我如何解释你的问题如下: 您有A1, A2, A3, A4B1, B2, B3, B4矩阵,所有矩阵的大小均为m x n。您希望形成4m x n矩阵,该矩阵由相应矩阵的逐元素乘法形成,在第一维中堆叠在一起。以下是如何做到这一点:

tf.concat_v2([A1*B1, A2*B2, A3*B3, A4*B4], 0)