ReichQin的博客

原创 TensorFlow2子类模型多输入多输出

在最近的一次项目中，因为需要模型具有多输入多输出，而且我的一个输出是一个包含张量的列表，所以无法使用函数式API或者容器去造模型，因为列表的添加操作不是一个层，而这两类的输出必须是层的结果，虽然可以用tf.keras.layers.Lambda将此操作变成层，但总归是牵强的，所以使用子类模型。 class Test(keras.Model): def __init__(self): super(Test, self).__init__() filters = 64