Python Tensorflow形状不匹配(WaveNet)

import tensorflow as tf

class WaveNet(tf.keras.Model):
    def __init__(self, num_layers=10, num_filters=32, kernel_size=2):
        super(WaveNet, self).__init__()
        self.layers = []
        for i in range(num_layers):
            dilation_rate = 2 ** i
            self.layers.append(tf.keras.layers.Conv1D(filters=num_filters, kernel_size=kernel_size, dilation_rate=dilation_rate, padding='causal'))
            self.layers.append(tf.keras.layers.Activation('relu'))
            self.layers.append(tf.keras.layers.Conv1D(filters=num_filters, kernel_size=1))
            self.layers.append(tf.keras.layers.BatchNormalization())
        self.output_layer = tf.keras.layers.Conv1D(filters=1, kernel_size=1)

    def call(self, inputs):
        x = inputs
        skips = []
        for layer in self.layers:
            x = layer(x)
            if isinstance(layer, tf.keras.layers.Conv1D) and layer.kernel_size == 2:
                skips.append(x)
        x = tf.add_n(skips)
        x = self.output_layer(x)
        return x

# 示例输入数据
batch_size = 32
time_steps = 1000
features = 1
input_data = tf.random.normal((batch_size, time_steps, features))

# 创建模型
model = WaveNet()

# 前向传播
output = model(input_data)
print(output.shape)  # 输出形状应为 (batch_size, time_steps, 1)