tensorflow2.0建模教程 常用操作集合系列2—Model自定义模型的设计(多层Layer嵌套成一个完整的计算网络模型)

推荐:
【tensorflow2.0最新建模教程】,如下:
tensorflow2.0教程 常用操作集合1—Layer自定义层的设计

常用操作集合系列2—Model自定义模型的设计

常用操作集合系列3—Model自定义模型的训练和参数调优loss gradient optimizer
常用操作集合系列4—损失函数正则化解决模型过拟合

在完成系列1自定义层设计后,下面我们把多层自定义神经网络嵌套成为一个完整的计算网络模型,代码示例如下:

#自定义层设计Mydense
Class Mydense(layer):
    def _init_(self,units):
        super(Mydense,self)._init_()
        self.units=uints
    def build(slef,input_shape):
        self.w=self.add_weight(shape=(input_shape[-1],units),initializer='random_normal',trainalbe=True)
        self.b=self.ad_weight(shape=uinits,initializer='random_normal',trainable=True)
    def call(self,input):
        return tf.matmul(input,self.w) 

猜你喜欢

转载自blog.csdn.net/weixin_37479258/article/details/100826635