tf.nn.conv1d
| Просмотреть исходный код на GitHub |
Вычисляет одномерную свёртку заданных 3-мерных тензоров входных и фильтрующих данных.
tf.nn.conv1d(
input, filters, stride, padding, data_format='NWC', dilations=None, name=None
)
Учитывая входной тензор формы batch_shape + [in_width, in_channels] если data_format является "NWC", или batch_shape + [in_channels, in_width] если data_format является "NCW", и тензор фильтра/ядра формы [filter_width, in_channels, out_channels], этот оператор изменяет форму аргументов, чтобы передать их в conv2d для выполнения эквивалентной операции свёртки.
Внутренне этот оператор изменяет форму входных тензоров и вызывает tf.nn.conv2d. Например, если data_format не начинается с "NC", тензор формы batch_shape + [in_width, in_channels] изменяется на форму batch_shape + [1, in_width, in_channels], а фильтр — на форму [1, filter_width, in_channels, out_channels]. Результат затем изменяется обратно на форму batch_shape + [out_width, out_channels] (где out_width является функцией шага и заполнения, как и в conv2d) и возвращается вызывающему методу.
| Аргументы | |
|---|---|
input | Тензор ранга не менее 3. Должен быть типа float16, float32, или float64. |
filters | Тензор ранга не менее 3. Должен иметь тот же тип, что и input. |
stride | Целое число или список ints, длина которого составляет 1 или 3. Количество элементов, на которые смещается фильтр вправо на каждом шаге. |
padding | 'SAME' или 'VALID' |
data_format | Необязательный string из "NWC", "NCW". По умолчанию — "NWC", данные хранятся в порядке batch_shape + [in_width, in_channels]. Формат "NCW" хранит данные как batch_shape + [in_channels, in_width]. |
dilations | Целое число или список ints, длина которого составляет 1 или 3, по умолчанию 1. Коэффициент расширения для каждой размерности входных данных. Если установлено значение k > 1, между каждым элементом фильтра по этой размерности будет k-1 пропущенных ячеек. Расширения по размерностям партии и глубины должны быть равны 1. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
А Tensor. Имеет тот же тип, что и вход. |
| Исключения | |
|---|---|
ValueError | если data_format является недопустимым. |
© 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/r2.3/api_docs/python/tf/nn/conv1d