keras包含各种内置层
tf.keras
包含各种内置层,例如:
- 卷积层:
Conv1D
,Conv2D
,Conv3D
,Conv2DTranspose
- 池化层:
MaxPooling1D
,MaxPooling2D
,MaxPooling3D
,AveragePooling1D
- RNN层:
GRU
,LSTM
,ConvLSTM2D
BatchNormalization
,Dropout
,Embedding
等
但是,如果找不到所需的内容,则可以通过创建自己的图层来扩展API。所有图层都继承了Layer
类并实现:
call
方法,它指定由图层完成的计算。build
方法,它创建图层的权重(这只是一种样式约定,因为您也可以在__init__
创建权重)。
要了解有关从头开始创建图层的更多信息,请阅读自定义图层和模型指南。
以下是tf.keras.layers.Dense
的基本实现:
【推荐】还在用 ECharts 开发大屏?试试这款永久免费的开源 BI 工具!
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步