Я пытаюсь передать пакет с различными начальными состояниями в свою пользовательскую ячейку RNN, чтобы иметь разные начальные состояния для каждого входного образца. В моей пользовательской ячейке RNN я указал для state_size значение (128, 8, 8), которое является формой каждого образца. Затем я передаю пакет начальных состояний в слой RNN, используя следующие строки:
RNNLayer = tf.keras.layers.RNN(cell, return_sequences=True, input_shape=x.shape, name='rnn')
x = RNNLayer(x, initial_state=[initial_state_batch])
Однако я сталкиваюсь со следующей ошибкой:
ValueError: An `initial_state` was passed that is not compatible with `cell.state_size`. Received `state_spec`=[InputSpec(shape=(None, 128, 8, 8), ndim=4)]; however `cell.state_size` is (128, 8, 8)
Кажется, что уровень RNN интерпретация пакета как одного исходного состояния.