十六、Keras 函数式 API

xlw_0579

已于 2024-02-22 10:46:37 修改

阅读量107

点赞数 1

分类专栏： Keras 文章标签：深度学习人工智能

于 2024-02-22 10:44:05 首次发布

本文链接：https://blog.csdn.net/xlw_6569/article/details/136227279

版权

Keras 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

Keras 函数式 API

from keras import Input, layers
input_tensor = Input(shape=(32,)) 
dense = layers.Dense(32, activation='relu') 
output_tensor = dense(input_tensor)

from keras.models import Sequential, Model
from keras import layers
from keras import Input
seq_model = Sequential() 
seq_model.add(layers.Dense(32, activation='relu', input_shape=(64,)))
seq_model.add(layers.Dense(32, activation='relu'))
seq_model.add(layers.Dense(10, activation='softmax'))
input_tensor = Input(shape=(64,)) 
x = layers.Dense(32, activation='relu')(input_tensor)
x = layers.Dense(32, activation='relu')(x)
output_tensor = layers.Dense(10, activation='softmax')(x)
model = Model(input_tensor, output_tensor)

model.summary() 
调用 model.summary() 的输出如下所示。
_________________________________________________________________
Layer (type) Output Shape Param # 
=================================================================
input_1 (InputLayer) (None, 64) 0 
_________________________________________________________________
dense_1 (Dense) (None, 32) 2080 
_________________________________________________________________
dense_2 (Dense) (None, 32) 1056 
_________________________________________________________________
dense_3 (Dense) (None, 10) 330 
=================================================================
Total params: 3,466
Trainable params: 3,466
Non-trainable params: 0
_________________________________________________________________