TensorFlow 2d Pooling，但只在1轴

Question

我正在创建一个SSL神经网络，我的输入张量是一个NxM张量，其中N是声波的长度，M是麦克风的数量。实际大小大约是14000x4

我需要池，但我只想池每列的行(而不是列在一起)。例如:

池(2)(张量)比;张量(N/2)xM

如果不把张量分成4个张量，形成4个单独的Pool1D，然后连接，这可能吗?

Pool1D给出维度错误Pool2D减少了行和列的数量

Answer 1

设置列的步长为1，

tf.keras.layers.MaxPooling2D(pool_size=(2, 2),strides=(2, 1), padding='same')

,

inputs = tf.random.normal(shape=(14000,4)) 
inputs = inputs[None,...,None] 
max_pool_2d = tf.keras.layers.MaxPooling2D(pool_size=(2, 2),strides=(2, 1), padding='same') 
max_pool_2d(inputs).shape
#[1, 7000, 4, 1]