Spec-Zone.ru › TensorFlow 2.9

tf.nn.conv1d

Просмотреть исходный код на GitHub

Вычисляет 1-мерную свёртку для тензоров входных и фильтрующих данных размерности 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 Имя операции (необязательно).
Возвращает
A Tensor. Имеет тот же тип, что и вход.
Исключения
ValueError если data_format недействителен.

© 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/nn/conv1d

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API