我正在使用 Keras 来构建我的架构。

我试图解决的回归问题对于不同的训练样本有不同的输出。

假设我的前两行y_train = [[16, 3], [6], ... ]对应于我的前两行X_train (input data)。我想将密集层单元的输出分配给这些y_train行的长度。

例如，对于第一个训练样本，我希望长度model.add(dense(2))为y_train[0]2，对于第二个训练样本，我希望长度model.add(dense(1))为y_train[0]1，依此类推。

我还认为我可以调用 Lambda 层而不是输出密集层，并且在该 Lambda 层中，我可以y_train使用 Lambda 层的 output_shape 参数包装每一行的长度并将其分配为输出单元。但是，我不知道如何y_train在模型的训练时访问我的模型，以便我可以使用y_train内部 lambda 层来实现这种情况。

有人可以帮我解决这个可变长度输出问题吗？