tensorflow中使用class构建神经网络的网络结构块

tensorflow中使用class构建神经网络的网络结构块

class用来声明网络结构

class MyModel(Model):    #MyModel表示声明神经网络的名字

            def __init__(self):

                               super(MyModel, self).__init__()   #初始化父类的参数

                                //初始化网络结构，构建神经网络所需的各种网络结构块

            def call(self, x):  #call()方法调用__init__()方法中完成初始化的网络结构块

                 y = self.d1(x)   #使用网络结构块

                 return y

 

代码演示：

#第一步，import
import tensorflow as tf #导入模块
from sklea