python的matmul_浅谈keras中的batch_dot,dot方法和TensorFlow的matmul

概述

在使用keras中的keras.backend.batch_dot和tf.matmul实现功能其实是一样的智能矩阵乘法，比如A,B,C,D,E,F,G,H，I,J,K,L都是二维矩阵，中间点表示矩阵乘法，AG 表示矩阵A 和G 矩阵乘法(A 的列维度等于G 行维度)，WX=Z

import keras.backend as K

import tensorflow as tf

import numpy as np

w = K.variable(np.random.randint(10,size=(10,12,4,5)))

k = K.variable(np.random.randint(10,size=(10,12,5,8)))

z = K.batch_dot(w,k)

print(z.shape) #(10, 12, 4, 8)

import keras.backend as K

import tensorflow as tf

import numpy as np

w = tf.Variable(np.random.randint(10,size=(10,12,4,5)),dtype=tf.float32)

k = tf.Variable(np.random.randint(10,size=(10,12,5,8)),dtype=tf.float32)

z = tf.matmul(w,k)

print(z.shape) #(10, 12, 4, 8)

示例

from keras import backend as K

<