乘法矩阵与向量(元素明智)Tensorflow

时间:2017-04-14 15:32:05

标签: tensorflow

我不确定将这个问题写成标题的方法。但是会在使用Tensorflow时需要帮助的事情上展示一个例子。

举个例子:

matrix_1 shape = [4,2]
matrix_2 shape  = [4,1]

matrix_1 * matrix 2
[[1,2],
 [3,4],
 [5,6],
 [7,8]]

 *

[[0.1],
 [0.2],
 [0.3],
 [0.4]]

 = [[0.1,0.2],
    [0.6,0.8],
    [1.5,1.8],
    [2.8,3.2]]

有没有算法来实现这个目标?

谢谢

这是我从上面的简化问题示例中得到的错误:

  

ValueError:尺寸必须相等,但对于' mul_13'为784和100。 (op:' Mul')输入形状:[100,784],[100]

1 个答案:

答案 0 :(得分:2)

标准tf.multiply(matrix_1, matrix_2)操作(或简写语法the decoded data is: undefined )将完全执行您在matrix_1 * matrix_2matrix_1上所需的计算。

但是,看起来您看到的错误消息是因为matrix_2具有形状matrix_2,而[100]必须[100, 1]才能获得elementwise broadcasting行为。使用tf.reshape(matrix_2, [100, 1])tf.expand_dims(matrix_2, 1)将其转换为正确的形状。

相关问题