Spec-Zone.ru › TensorFlow 2.9

tf.nn.dilation2d

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

Вычисляет градационную дилатацию тензоров размерности 4-D input и 3-D filters.

tf.nn.dilation2d(
    input, filters, strides, padding, data_format, dilations, name=None
)

Тензор input имеет форму [batch, in_height, in_width, depth], а тензор filters имеет форму [filter_height, filter_width, depth], то есть каждый входной канал обрабатывается независимо от других с собственной структурирующей функцией. Тензор output имеет форму [batch, out_height, out_width, depth]. Пространственные размеры тензора вывода зависят от алгоритма padding. В настоящее время мы поддерживаем только по умолчанию "NHWC" data_format.

Подробно, градационная морфологическая дилатация 2-D — это корреляция max-sum (для согласованности с conv2d, мы используем неотраженные фильтры):

output[b, y, x, c] =
   max_{dy, dx} input[b,
                      strides[1] * y + rates[1] * dy,
                      strides[2] * x + rates[2] * dx,
                      c] +
                filters[dy, dx, c]

Максимальный пул является частным случаем, когда размер фильтра равен размеру ядра пула и содержит все нули.

Примечание о двойственности: Дилатация input с помощью filters равна отрицанию эрозии -input с помощью отраженного filters.

Аргументы
input А Tensor. Должен быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64. 4-D с формой [batch, in_height, in_width, depth].
filters А Tensor. Должен иметь тот же тип, что и input. 3-D с формой [filter_height, filter_width, depth].
strides Список ints, длина которого равна >= 4. Шаг скользящего окна для каждого измерения входного тензора. Должен быть: [1, stride_height, stride_width, 1].
padding А string из: "SAME", "VALID". Тип алгоритма заполнения. Подробнее см. здесь.
data_format А string, в настоящее время поддерживается только "NHWC" .
dilations Список ints, длина которого равна >= 4. Шаг входа для дилатации атроу морфологической дилатации. Должен быть: [1, rate_height, rate_width, 1].
name Имя операции (необязательно).
Возвращаемое значение
А Tensor. Имеет тот же тип, что и input.

© 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/dilation2d

Spec-Zone.ru

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