Keras中的矩阵乘法

时间:2017-05-09 13:44:45

标签: python keras

我尝试使用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)

我知道这不起作用,但我也不知道如何让它发挥作用。

1 个答案:

答案 0 :(得分:16)

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

def create_permutations_auto(img):
    colors = {0: 'red', 1: 'green', 2: 'blue'}
    index_perm = list(itertools.permutations(colors.keys()))
    cp = [tuple(colors[i] for i in perm) for perm in index_perm] 
    im_list = [np.dstack([img[:, :, i] for i in perm]) for perm in index_perm]
    return cp, im_list