tf.contrib.rnn.ConvLSTMCell
Сверточная ячейка LSTM.
Наследуется от: RNNCell
tf.contrib.rnn.ConvLSTMCell(
conv_ndims, input_shape, output_channels, kernel_shape, use_bias=True,
skip_connection=False, forget_bias=1.0, initializers=None, name='conv_lstm_cell'
)
https://arxiv.org/pdf/1506.04214v1.pdf
| Аргументы | |
|---|---|
conv_ndims | Разность размерностей свертки (1, 2 или 3). |
input_shape | Форма входных данных в виде кортежа целых чисел, без учёта размера батча. |
output_channels | Целое число, количество выходных каналов сверточной ячейки LSTM. |
kernel_shape | Форма ядра в виде кортежа целых чисел (размерности 1, 2 или 3). |
use_bias | (логическое значение) Использовать смещение в свертках. |
skip_connection | Если установлено в True, объединить входные данные с выходом сверточной ячейки LSTM. По умолчанию False. |
forget_bias | Смещение для забывания. |
initializers | Не используется. |
name | Имя модуля. |
| Исключения | |
|---|---|
ValueError | Если skip_connection равно True и шаг отличается от 1, или если input_shape несовместимо с conv_ndims. |
| Атрибуты | |
|---|---|
graph | УСТАРЕВШАЯ ФУНКЦИЯ |
output_size | Целое число или TensorShape: размер выходных данных, производимых этой ячейкой. |
scope_name | |
state_size | Размерность состояний, используемых этой ячейкой. Может быть представлено целым числом, TensorShape или кортежем целых чисел или TensorShape. |
Методы
get_initial_state
get_initial_state(
inputs=None, batch_size=None, dtype=None
)
zero_state
zero_state(
batch_size, dtype
)
Возвращает тензор(ы) состояния, заполненные нулями.
| Аргументы | |
|---|---|
batch_size | Целое число, число с плавающей точкой или тензор, представляющий размер батча. |
dtype | Тип данных для состояния. |
| Возвращает | |
|---|---|
Если state_size — целое число или TensorShape, то значение возврата — тензор N-D формы [batch_size, state_size] заполненный нулями. Если |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/rnn/ConvLSTMCell