tensorflow::ops::FusedPadConv2D
#include <nn_ops.h>
Выполняет заполнение как предобработку во время свёртки.
Краткое описание
Аналогично FusedResizeAndPadConv2d, этот оператор позволяет оптимизированную реализацию, где стадия пространственного заполнения объединена со стадией поиска im2col, но в этом случае без билинейной фильтрации, необходимой для изменения размера. Объединение заполнения позволяет избежать необходимости записи промежуточных результатов как целых тензоров, что уменьшает нагрузку на память, и мы можем получить некоторые преимущества за счёт задержки, объединив вычисления преобразований. Атрибут data_format для Conv2D не поддерживается этим оператором, и вместо него используется порядок 'NHWC'. Внутри этот оператор использует единственный кэш-буфер на график, что означает, что он будет блокироваться, если выполняются несколько версий параллельно. Это потому, что этот оператор в первую очередь является оптимизацией для минимизации использования памяти.
Аргументы:
- scope: Объект Scope
- input: 4-мерный с формой
[batch, in_height, in_width, in_channels]. - paddings: Матрица с двумя столбцами, определяющая размеры заполнения. Количество строк должно быть таким же, как ранг
input. - filter: 4-мерный с формой
[filter_height, filter_width, in_channels, out_channels]. - strides: 1-мерный массив длиной 4. Шаг скользящего окна для каждого измерения
input. Должен быть в том же порядке, что и измерение, указанное в формате. - padding: Тип алгоритма заполнения для использования.
Возвращает:
-
Output: Тензор вывода.
| Конструкторы и деструкторы | |
|---|---|
FusedPadConv2D(const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) |
| Общедоступные атрибуты | |
|---|---|
operation | |
output | |
| Общедоступные функции | |
|---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const | |
Общедоступные атрибуты
operation
Operation operation
output
::tensorflow::Output output
Общедоступные функции
FusedPadConv2D
FusedPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
© 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/cc/class/tensorflow/ops/fused-pad-conv2-d