tf.keras.layers.ConvLSTM2D
| Просмотреть исходный код на GitHub |
2D Свёрточный LSTM.
Наследуется от: RNN, Layer, Module
tf.keras.layers.ConvLSTM2D(
filters,
kernel_size,
strides=(1, 1),
padding='valid',
data_format=None,
dilation_rate=(1, 1),
activation='tanh',
recurrent_activation='hard_sigmoid',
use_bias=True,
kernel_initializer='glorot_uniform',
recurrent_initializer='orthogonal',
bias_initializer='zeros',
unit_forget_bias=True,
kernel_regularizer=None,
recurrent_regularizer=None,
bias_regularizer=None,
activity_regularizer=None,
kernel_constraint=None,
recurrent_constraint=None,
bias_constraint=None,
return_sequences=False,
return_state=False,
go_backwards=False,
stateful=False,
dropout=0.0,
recurrent_dropout=0.0,
**kwargs
)
Аналогичен слою LSTM, но преобразования входных данных и рекуррентные преобразования — оба свёрточные.
| Аргументы | |
|---|---|
filters | Целое число, размерность выходного пространства (то есть количество выходных фильтров в свёрточном слое). |
kernel_size | Целое число или кортеж/список из n целых чисел, определяющих размеры окна свёртки. |
strides | Целое число или кортеж/список из n целых чисел, определяющих шаги свёртки. Указание любого значения шага != 1 несовместимо с указанием любого значения dilation_rate != 1. |
padding | Одно из "valid" или "same" (регистр игнорируется). "valid" означает отсутствие заполнения. "same" приводит к добавлению заполнения равномерно слева/справа или сверху/снизу входных данных таким образом, чтобы выходные данные имели такую же высоту/ширину, как и входные. |
data_format | Строка, одно из channels_last (по умолчанию) или channels_first. Порядок измерений ввода. channels_last соответствует входным данным с формой (batch, time, ..., channels), а channels_first соответствует входным данным с формой (batch, time, channels, ...). По умолчанию используется значение image_data_format в файле конфигурации Keras по адресу ~/.keras/keras.json. Если вы никогда его не устанавливали, то он будет "channels_last". |
dilation_rate | Целое число или кортеж/список из n целых чисел, определяющих скорость расширения для свёртки с расширением. В настоящее время указание любого значения dilation_rate != 1 несовместимо с указанием любого значения strides != 1. |
activation | Функция активации для применения. По умолчанию используется функция активации гиперболического тангенса (tanh(x)). |
recurrent_activation | Функция активации для использования на рекуррентном шаге. |
use_bias | Булево значение, указывает, использует ли слой вектор смещения. |
kernel_initializer | Инициализатор для матрицы весов kernel, используемой для линейного преобразования входов. |
recurrent_initializer | Инициализатор для матрицы весов recurrent_kernel, используемой для линейного преобразования рекуррентного состояния. |
bias_initializer | Инициализатор для вектора смещения. |
unit_forget_bias | Булево значение. Если True, добавляет 1 к смещению забывающего блока при инициализации. Используется в сочетании с bias_initializer="zeros". Рекомендуется в Jozefowicz et al., 2015 |
kernel_regularizer | Функция регуляризации, применяемая к матрице весов kernel . |
recurrent_regularizer | Функция регуляризации, применяемая к матрице весов recurrent_kernel . |
bias_regularizer | Функция регуляризации, применяемая к вектору смещения. |
activity_regularizer | Функция регуляризации, применяемая к. |
kernel_constraint | Функция ограничения, применяемая к матрице весов kernel . |
recurrent_constraint | Функция ограничения, применяемая к матрице весов recurrent_kernel . |
bias_constraint | Функция ограничения, применяемая к вектору смещения. |
return_sequences | Булево значение. Вернуть ли только последний выход в последовательности вывода, или всю последовательность. (по умолчанию False) |
return_state | Булево значение. Вернуть ли последнее состояние дополнительно к выводу. (по умолчанию False) |
go_backwards | Булево значение (по умолчанию False). Если True, обрабатывать последовательность ввода в обратном порядке. |
stateful | Булево значение (по умолчанию False). Если True, последнее состояние для каждого образца с индексом i в пакете будет использоваться в качестве начального состояния для образца с индексом i в следующем пакете. |
dropout | Число с плавающей запятой от 0 до 1. Доля единиц, которые нужно выбросить для линейного преобразования входов. |
recurrent_dropout | Число с плавающей запятой от 0 до 1. Доля единиц, которые нужно выбросить для линейного преобразования рекуррентного состояния. |
Аргументы вызова:
-
inputs: 5D тензор. -
mask: Бинарный тензор формы(samples, timesteps), указывающий, следует ли маскировать определённый шаг во времени. -
training: Булево значение Python, указывающее, должен ли слой вести себя в режиме обучения или в режиме вывода. Этот аргумент передаётся в ячейку при её вызове. Это актуально только еслиdropoutилиrecurrent_dropoutустановлены. -
initial_state: Список начальных тензоров состояния, которые должны быть переданы в первый вызов ячейки.
Форма входных данных: - Если data_format='channels_first' 5D тензор с формой: (samples, time, channels, rows, cols) - Если data_format='channels_last' 5D тензор с формой: (samples, time, rows, cols, channels) Форма вывода:
- Если
return_state: список тензоров. Первый тензор — выходные данные. Остальные тензоры — последние состояния, каждый 4D тензор с формой:(samples, filters, new_rows, new_cols)если data_format='channels_first' или формой:(samples, new_rows, new_cols, filters)если data_format='channels_last'.rowsиcolsзначения могут измениться из-за заполнения. - Если
return_sequences: 5D тензор с формой:(samples, timesteps, filters, new_rows, new_cols)если data_format='channels_first' или формой:(samples, timesteps, new_rows, new_cols, filters)если data_format='channels_last'. - В противном случае, 4D тензор с формой:
(samples, filters, new_rows, new_cols)если data_format='channels_first' или формой:(samples, new_rows, new_cols, filters)если data_format='channels_last'.
| Исключения | |
|---|---|
ValueError | в случае неверных аргументов конструктора. |
Ссылки:
- Shi et al., 2015 (в текущей реализации петля обратной связи по выводу ячеек не включена).
| Атрибуты | |
|---|---|
activation | |
bias_constraint | |
bias_initializer | |
bias_regularizer | |
data_format | |
dilation_rate | |
dropout | |
filters | |
kernel_constraint | |
kernel_initializer | |
kernel_regularizer | |
kernel_size | |
padding | |
recurrent_activation | |
recurrent_constraint | |
recurrent_dropout | |
recurrent_initializer | |
recurrent_regularizer | |
states | |
strides | |
unit_forget_bias | |
use_bias | |
Методы
reset_states
reset_states(
states=None
)
Сбросить записанные состояния для слоёв RNN со с сохранением состояния.
Может быть использован только, когда слой RNN был сконструирован с stateful = True. Args: states: Массивы NumPy, содержащие значение для начального состояния, которое будет передано ячейке в первый момент времени. Если значение None, будет создан массив NumPy с нулями, основанный на размере состояния ячейки.
| Исключения | |
|---|---|
AttributeError | Когда слой RNN не имеет сохранения состояния. |
ValueError | Когда размер пакета для слоя RNN неизвестен. |
ValueError | Когда массив NumPy несовместим со состоянием слоя RNN, либо по размеру, либо по типу данных. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/layers/ConvLSTM2D