Keras在未知大小的轴上分裂张量

Question

我有一个Keras张量tens，形状（无，无，10）（包括批大小）。我想将其拆分为轴1上的形状张量（无，10）的列表。如果我知道轴1的大小为k，我可以做

layers.Lambda(lambda x: [x[:, i, :] for i in range(x.shape[1])])(tens)

但是，由于大小未知，因此在构建此模型时出现错误__index__ returned non-int (type NoneType)。我该如何解决？