python - Keras 中的矩阵乘法

Question

我尝试使用 Keras 在 python 程序中将两个矩阵相乘。

import keras.backend as K
import numpy as np
A = np.random.rand(10,500)
B = np.random.rand(500,6000)

x = K.placeholder(shape=A.shape)
y = K.placeholder(shape=B.shape)
xy = K.dot(x, y)

xy.eval(A,B)

我知道这行不通，但我也不知道如何让它发挥作用。

Answer 1

您需要使用变量而不是占位符。

import keras.backend as K
import numpy as np
A = np.random.rand(10,500)
B = np.random.rand(500,6000)

x = K.variable(value=A)
y = K.variable(value=B)

z = K.dot(x,y)

# Here you need to use K.eval() instead of z.eval() because this uses the backend session
K.eval(z)