据我所知,tf.layers.conv1d使用如下管道:BatchToSpaceND-> conv1d-> SpaceToBatchND。那么问题是如何从管道中删除(或禁用)BatchToSpaceND和SpaceToBatchND?
答案 0 :(得分:0)
正如我所研究的那样,在不更改和重建张量流源代码的情况下,不可能从tf.layers.conv1d中删除BatchToSpaceND和SpaceToBatchND。解决方案之一是用tf.nn.conv1d替换层,这是卷积层的低级表示(实际上tf.layers.conv1d是tf.nn.conv1d的包装)。这些低层实现不包括BatchToSpaceND和SpaceToBatchND。