问如何在Keras中按列拆分张量以实现STFCN
EN

Stack Overflow用户

提问于 2017-01-05 19:52:22

回答 0查看 4.6K关注 0票数 5

我想在Keras中实现时空完全卷积网络(STFCN)。我需要将3D卷积输出的每个深度列，例如具有形状(64, 16, 16)的张量，作为单独的LSTM的输入。

为了说明这一点，我有一个维度(channels, height, width)的(64 x 16 x 16)张量。我需要(显式或隐式地)将张量拆分为16 * 16 = 256个形状为(64 x 1 x 1)的张量。

这是STFCN论文中的一个图表，用来说明时空模块。我上面描述的是“空间特征”和“时空模块”之间的箭头。

如何在Keras中最好地实现这个想法？

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/41484413

复制

相似问题

问如何在Keras中按列拆分张量以实现STFCNEN