如何在整个矩阵上应用Conv1D层

时间:2020-04-23 11:28:01

标签: tensorflow keras

我有一个矩阵,每行代表一个坐标为(x,y,z)的点。我想使用3个共享的MLP层(64,128,1024)(内核大小为1的Conv1D)并在最后,我想使用MaxPooling1D汇总功能。 我的问题是如何定义我的输入是整个矩阵(我的意思是我希望每一层都应用在矩阵的整个行上,而不仅仅是行上) 我编写了代码,但我确定这是错误的

**Model=Sequential([
                    Conv1D(64,1,input_dim=(1,3),activation='relu')
                    BatchNormalization(axis=-1)
                    Conv1D(128,1,activation='relu')
                    BatchNormalization(axis=-1)
                    Conv1D(1021,1,activation='relu')
                    BatchNormalization(axis=-1)
                    MaxPooling1D(1)

  ])**

预先感谢

0 个答案:

没有答案