如何在Keras中实现具有动态形状的自定义输出层？

from tensorflow.keras.layers import Layer, Lambda
import tensorflow.keras.backend as K

class DynamicShapeLayer(Layer):
    def __init__(self, **kwargs):
        super(DynamicShapeLayer, self).__init__(**kwargs)

    def call(self, inputs):
        input_shape = K.int_shape(inputs)
        output_shape = (input_shape[0], input_shape[1] + 1)  # 动态计算输出形状
        output = K.zeros(output_shape)  # 初始化输出
        return output

    def compute_output_shape(self, input_shape):
        return (input_shape[0], input_shape[1] + 1)  # 返回输出形状

# 使用Lambda层将自定义操作封装为一个层
dynamic_shape_layer = Lambda(lambda x: DynamicShapeLayer()(x))

# 构建模型
input_layer = Input(shape=(10,))
output_layer = dynamic_shape_layer(input_layer)
model = Model(inputs=input_layer, outputs=output_layer)