python - 有没有办法从 tf.layers.conv1d 中删除 BatchToSpaceND？

Question

据我所知，tf.layers.conv1d 使用这样的管道：BatchToSpaceND -> conv1d -> SpaceToBatchND。所以问题是如何从管道中删除（或禁用）BatchToSpaceND 和 SpaceToBatchND？

标签： pythontensorflowconv-neural-network

Answer 1

正如我所调查的那样，如果不更改和重建 tensorflow 源代码，就不可能从 tf.layers.conv1d 中删除 BatchToSpaceND 和 SpaceToBatchND。解决方案之一是将层替换为 tf.nn.conv1d，这是卷积层的低级表示（实际上 tf.layers.conv1d 是 tf.nn.conv1d 的包装器）。这些低级实现不包括 BatchToSpaceND 和 SpaceToBatchND。