标签: python tensorflow
我有一个形状为[?,6,2]的输入张量,并且窗口大小为3。我想沿着输入滑动窗口,将它们连接起来,并通过一个密集的层运行,该层需要6个输入并且有1个输出。
它将在索引[0-2],[1-3],[2-4],[3-5]上运行串联的输入值,并将密集的网络输出串联起来以创建大小为4的新张量
所以[?,6,2]-> [?,4]
完成此任务的最佳方法是什么?
我觉得可以用conv2d做到这一点,但不确定如何制定它。