Trax中的MaxPooling一维层

Question

我如何在google中编码我自己的MaxPooling_1D层？我理解当前的最大池是基于2D最大池的。

下面是我如何尝试使用Keras 1d层

   import trax.layers as tl
   def computePool(max_pool_1d,in_tensor):
      #print(in_tensor)
      return max_pool_1d(in_tensor)
   def maxPooling1D():
      max_pool_1d = tf.keras.layers.GlobalMaxPooling1D()
      return tl.Fn('maxPooling1D', lambda x: computePool(max_pool_1d, x))

但这在模型中是不会被接受的。我想要创建一个池大小=2的层

Answer 1

答案是

tl.MaxPool(pool_size=(2,), strides=(1,), padding='SAME'),