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