keras包含各种内置层

tf.keras包含各种内置层,例如:

  • 卷积层: Conv1DConv2DConv3DConv2DTranspose
  • 池化层: MaxPooling1DMaxPooling2DMaxPooling3DAveragePooling1D
  • RNN层: GRULSTMConvLSTM2D
  • BatchNormalizationDropoutEmbedding

但是,如果找不到所需的内容,则可以通过创建自己的图层来扩展API。所有图层都继承了Layer类并实现:

  • call方法,它指定由图层完成的计算。
  • build方法,它创建图层的权重(这只是一种样式约定,因为您也可以在__init__创建权重)。

要了解有关从头开始创建图层的更多信息,请阅读自定义图层和模型指南。

以下是tf.keras.layers.Dense的基本实现:

原文地址:https://www.cnblogs.com/DDBD/p/13706636.html