如何使喀拉拉邦中的图层块返回比张量大的图层？

Question

我想在由conv2d + batchnorm + relu + maxpool组成的keras中制作一个图层块。通常我会定义一个函数

def conv_block(x):
    x = Conv2D(...)(x)
    x = BatchNormalization(...)(x)
    x = Activation('relu')(x)
    x = MaxPool2D(...)(x)
    return x

但是这将返回张量。我想返回一个图层对象，因为我想使用TimeDistributed将此4层操作应用于数据序列，而不是将TimeDistributed应用于每个单独的图层。我该怎么办？