python计算两个正整数乘积_在keras中创建一个神经网络来将两个输入整数相乘

我在pythonv2.7(Tensorflow后端)中使用kerasv2.0.8来创建计算简单算术函数(加法、减法、乘法等)的小型神经网络，但我有点困惑。下面的代码是我的网络，它生成带有相应标签的整数的随机训练数据集(两个输入相加)：def create_data(low, high, examples):

train_data = []

label_data = []

a = np.random.randint(low=low, high=high, size=examples, dtype='int')

b = np.random.randint(low=low, high=high, size=examples, dtype='int')

for i in range(0, examples):

train_data.append([a[i], b[i]])

label_data.append((a[i] + b[i]))

train_data = np.array(train_data)

label_data = np.array(label_data)

return train_data, label_data

X, y = create_data(0, 500, 10000)

model = Sequential()

model.add(Dense(3, input_dim=2))

model.add(Dense(5, activation='relu'))

model.add(Dense(3, activation='relu'))

mod