tf.raw_ops.FusedPadConv2D
Выполняет заполнение как предобработку во время свёртки.
tf.raw_ops.FusedPadConv2D(
input, paddings, filter, mode, strides, padding, name=None
)
Подобно FusedResizeAndPadConv2d, этот оператор позволяет оптимизированную реализацию, где стадия пространственного заполнения интегрирована с запросом im2col, но в данном случае без билинейной фильтрации, необходимой для изменения размера. Интегрирование заполнения предотвращает необходимость записи промежуточных результатов как полных тензоров, уменьшая потребность в памяти, и мы можем получить некоторые преимущества задержки, объединив вычисления преобразований. Атрибут data_format для Conv2D не поддерживается этим оператором, и вместо этого используется порядок «NHWC». Внутренне этот оператор использует один общий кеш на граф, что означает, что он будет блокировать, если параллельно выполняются несколько версий. Это связано с тем, что данный оператор в первую очередь является оптимизацией для минимизации потребления памяти.
| Аргументы | |
|---|---|
input | A Tensor. Должен быть одним из следующих типов: half, float32, float64. 4-мерный с формой [batch, in_height, in_width, in_channels]. |
paddings | A Tensor типа int32. Матрица из двух столбцов, определяющая размеры заполнения. Количество строк должно совпадать с рангом input. |
filter | A Tensor. Должен иметь тот же тип, что и input. 4-мерный с формой [filter_height, filter_width, in_channels, out_channels]. |
mode | A string из: "REFLECT", "SYMMETRIC". |
strides | Список ints. 1-мерный список длины 4. Шаг скользящего окна для каждого измерения input. Должен быть в том же порядке, что и измерение, указанное в формате. |
padding | A string из: "SAME", "VALID". Тип алгоритма заполнения, который необходимо использовать. |
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.3/api_docs/python/tf/raw_ops/FusedPadConv2D