有什么方法可以从tf.layers.conv1d中删除BatchToSpaceND?

时间:2020-02-05 17:03:07

标签: python tensorflow conv-neural-network

据我所知,tf.layers.conv1d使用如下管道:BatchToSpaceND-> conv1d-> SpaceToBatchND。那么问题是如何从管道中删除(或禁用)BatchToSpaceND和SpaceToBatchND?

1 个答案:

答案 0 :(得分:0)

正如我所研究的那样,在不更改和重建张量流源代码的情况下,不可能从tf.layers.conv1d中删除BatchToSpaceND和SpaceToBatchND。解决方案之一是用tf.nn.conv1d替换层,这是卷积层的低级表示(实际上tf.layers.conv1d是tf.nn.conv1d的包装)。这些低层实现不包括BatchToSpaceND和SpaceToBatchND。