machine-learning - 如何将未知大小的时间步输入到 TimeDistributed Keras 层

Question

我的输入状态形状= (84,84,4)

state = Input(shape=(84,84,4), dtype="float")

所以我想将其传递给时间步长大小=1..5(范围为1到5)的某个TimeDistributed层，但我不知道它到底等于哪个。

我的下一层是这样的:

conv1 = TimeDistributed(Convolution2D(16, 8, 8, subsample=(4, 4), border_mode='valid',
                                      activation='relu', dim_ordering='tf'))(state)

我在这一层遇到了错误:

IndexError: tuple index out of range

我只想传递一个未知的时间序列大小到 TimeDistributed，然后再传递到 LSTM。

Answer 1

所以基本上在 Keras 中 - 您需要提供序列长度，因为在计算过程中 Keras 层接受具有指定形状的输入 numpy 数组 - 这使得所有输入都强制(至少在一批中) )以固定长度。但是 - 您仍然可以通过 0 填充来处理不同的输入大小(通过在开头添加所有零虚拟时间步来使所有序列大小相等)，然后 masking是什么让你的网络相当于一个不同长度的输入网络。