tf.compat.v1.nn.dilation2d
Вычисляет серое расширение 4-D тензоров input и 3-D тензоров filter.
tf.compat.v1.nn.dilation2d(
input,
filter=None,
strides=None,
rates=None,
padding=None,
name=None,
filters=None,
dilations=None
)
Тензор input имеет форму [batch, in_height, in_width, depth], а тензор filter имеет форму [filter_height, filter_width, depth]; это означает, что каждый входной канал обрабатывается независимо от остальных с помощью собственной структурирующей функции. Тензор output имеет форму [batch, out_height, out_width, depth]. Пространственные размеры выходного тензора зависят от алгоритма padding. В настоящее время поддерживается только формат по умолчанию «NHWC» data_format.
Подробно, серое морфологическое двумерное расширение — это корреляция с максимумом суммы (для согласованности с conv2d используются немиррорные фильтры):
output[b, y, x, c] =
max_{dy, dx} input[b,
strides[1] * y + rates[1] * dy,
strides[2] * x + rates[2] * dx,
c] +
filter[dy, dx, c]
Пулинг с максимумом — это частный случай, когда размер фильтра равен размеру ядра пулинга, а фильтр содержит только нули.
Примечание о двойственности: расширение input элементом filter равно отрицанию эрозии -input отражённым элементом filter.
| Аргументы | |
|---|---|
input | Тензор Tensor. Должен иметь один из следующих типов: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64. Тензор 4-D с формой [batch, in_height, in_width, depth]. |
filter | Тензор Tensor. Должен иметь тот же тип, что и input. Тензор 3-D с формой [filter_height, filter_width, depth]. |
strides | Список значений ints длиной >= 4. Шаг скользящего окна для каждого измерения входного тензора. Должен быть: [1, stride_height, stride_width, 1]. |
rates | Список значений ints длиной >= 4. Шаг входных данных для атрусного морфологического расширения. Должен быть: [1, rate_height, rate_width, 1]. |
padding | Значение string из: "SAME", "VALID". Тип используемого алгоритма дополнения. |
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/api_docs/python/tf/compat/v1/nn/dilation2d