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