我想让神经网络输出3个不同的矩阵,两个是形状为(8,1)的矩阵,另一个是形状为(12,1)的矩阵。使用Keras可以实现吗?
回答:
当然可以,您可以在倒数第二层之后添加多个具有所需形状的层。然后将这些输出层作为列表传递给model.compile(..., outputs=[out1, out2, ...])
。您需要使用函数式API来实现这一点,而不能使用Sequential
模型。
我想让神经网络输出3个不同的矩阵,两个是形状为(8,1)的矩阵,另一个是形状为(12,1)的矩阵。使用Keras可以实现吗?
回答:
当然可以,您可以在倒数第二层之后添加多个具有所需形状的层。然后将这些输出层作为列表传递给model.compile(..., outputs=[out1, out2, ...])
。您需要使用函数式API来实现这一点,而不能使用Sequential
模型。