tensorflow自定义网络结构自定义网络层自定义模型类
自定义网络层自定义层需要继承tf.keras.layers.Layer类,重写init,build,call__init__,执行与输入无关的初始化build,了解输入张量的形状,定义需要什么输入call,进行正向计算class MyDense(tf.keras.layers.Layer): def __init__(self,units): # units 神经...