python - 将 Keras 张量拆分为列表

Question

我有一个大小为 [61,80,?] 的 keras 张量，带有“？” 用于自定义层中的批量大小。当我尝试使用 tf.unstack 将此张量作为大小 [61,80] 的 2D 张量列表解开时，我收到一个错误 -

ValueError：无法从形状推断 num (61, 80, ?)

由于没有 keras.backend.unstack，请建议如何执行此操作。

标签： pythontensorflowkeraskeras-layer