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