tensorflow::ops::FusedPadConv2D
#include <nn_ops.h>Выполняет заполнение в качестве предварительной обработки во время свертки.
Краткое содержание
Подобно FusedResizeAndPadConv2d, эта операция позволяет оптимизировать реализацию, в которой этап преобразования пространственного заполнения объединен с поиском im2col, но в данном случае без билинейной фильтрации, необходимой для изменения размера. Объединение заполнения предотвращает необходимость записи промежуточных результатов в виде целых тензоров, снижая нагрузку на память, и мы можем получить некоторую экономию времени за счет объединения вычислений преобразования. Атрибут data_format для Conv2D не поддерживается этой операцией, и вместо него используется порядок 'NHWC'. Внутри эта операция использует один буфер для временных данных на каждый граф, что означает, что она будет блокироваться, если несколько версий выполняются параллельно. Это связано с тем, что этот оператор в первую очередь является оптимизацией для минимизации использования памяти.
Аргументы:
- scope: Объект Scope
- Входные данные: 4D с формой
[batch, in_height, in_width, in_channels]. - paddings: Двухстолбцовая матрица, определяющая размеры отступов. Количество строк должно совпадать с рангом
input. - фильтр: 4D с формой
[filter_height, filter_width, in_channels, out_channels]. - strides: 1-D длиной 4. Шаг скользящего окна для каждого измерения
input. Должен быть в том же порядке, что и измерение, указанное в формате. - Заполнение: Тип используемого алгоритма заполнения.
Возвращает:
-
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
выход
::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 )
узел
::tensorflow::Node * node() const
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Output
operator::tensorflow::Output() const