tf.raw_ops.FusedPadConv2D
Выполняет заполнение (padding) как предподготовку во время свёртки.
tf.raw_ops.FusedPadConv2D(
input, paddings, filter, mode, strides, padding, name=None
)
Аналогично FusedResizeAndPadConv2d, этот оператор позволяет использовать оптимизированную реализацию, где стадия пространственного заполнения (padding) объединяется со стадией поиска im2col, но в данном случае без билинейной фильтрации, необходимой для изменения размера. Объединение заполнения (padding) предотвращает необходимость записи промежуточных результатов в виде целых тензоров, что снижает нагрузку на память, и мы можем получить некоторые преимущества в скорости, объединив вычисления преобразований. Атрибут data_format для Conv2D не поддерживается этим оператором, и вместо него используется порядок «NHWC». Внутренне этот оператор использует один общий буфер scratch для каждой графы, что означает, что он заблокируется, если выполняются несколько версий параллельно. Это связано с тем, что этот оператор в первую очередь представляет собой оптимизацию для минимизации использования памяти.
| Аргументы | |
|---|---|
input | A Tensor. Должен быть одним из следующих типов: half, bfloat16, 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". Тип алгоритма заполнения (padding) для использования. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
A 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/raw_ops/FusedPadConv2D