Tensorflow Python:如何在张量上连续应用密集层

时间:2019-08-27 06:20:28

标签: python tensorflow

我有一个形状为[?,6,2]的输入张量,并且窗口大小为3。我想沿着输入滑动窗口,将它们连接起来,并通过一个密集的层运行,该层需要6个输入并且有1个输出。

它将在索引[0-2],[1-3],[2-4],[3-5]上运行串联的输入值,并将密集的网络输出串联起来以创建大小为4的新张量

所以[?,6,2]-> [?,4]

完成此任务的最佳方法是什么?

我觉得可以用conv2d做到这一点,但不确定如何制定它。

0 个答案:

没有答案