如何在TensowFlow中对1维ConvNet(conv1d)执行最大池化？

Question

我正在训练文本卷积神经网络(在角色级别),我想做最大化池.tf.nn.max_pool期望等级4 Tensor,但是1-d convnets在tensorflow中是等级3([批次,宽度,深度]),所以当我将conv1d的输出传递给max pool函数时,这是错误:

ValueError: Shape (1, 144, 512) must have rank 4

我对一般的张量流和深度学习框架不熟悉,并希望就这里的最佳实践提出建议,因为我可以想象有多种解决方法.如何在1-d情况下执行max-pooling？

谢谢.

Answer 1

一个快速的方法是添加一个额外的单一维度,即制作形状(1,1,144,512),从那里你可以用tf.squeeze将其缩小.

我对其他方法感到好奇.