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
© 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/r1.15/api_docs/cc/class/tensorflow/ops/fused-pad-conv2-d