tensorflow::ops::FusedPadConv2D
#include <nn_ops.h>
Выполняет дополнение как предварительную обработку во время свёртки.
Summary
Аналогично 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/r2.3/api_docs/cc/class/tensorflow/ops/fused-pad-conv2-d