тензорный поток:: опс:: FusedResizeAndPadConv2D
#include <nn_ops.h>
Выполняет изменение размера и заполнение в качестве предварительной обработки во время свертки.
Краткое содержание
Часто можно более эффективно выполнять пространственные преобразования на этапе упаковки свертки, поэтому эта операция позволяет оптимизировать реализацию, в которой эти этапы объединяются вместе. Это избавляет от необходимости записывать промежуточные результаты в виде целых тензоров, уменьшая нагрузку на память, и мы можем получить некоторый выигрыш в задержке за счет объединения вычислений преобразования. Атрибут data_format для Conv2D не поддерживается этой операцией и по умолчанию имеет порядок «NHWC». Внутри эта операция использует один рабочий буфер для каждого графа, что означает, что она будет блокироваться, если несколько версий выполняются параллельно. Это связано с тем, что этот оператор в первую очередь предназначен для оптимизации использования памяти.
Аргументы:
- область: объект области.
- ввод: 4-D с формой
[batch, in_height, in_width, in_channels]
. - размер: 1-D int32 Тензор из 2 элементов:
new_height, new_width
. Новый размер изображений. - отступы: матрица из двух столбцов, определяющая размеры отступов. Количество строк должно совпадать с рангом
input
. - фильтр: 4-D с формой
[filter_height, filter_width, in_channels, out_channels]
. - шаги: 1-D длиной 4. Шаг скользящего окна для каждого измерения
input
. Должен быть в том же порядке, что и размер, указанный в формате. - дополнение: тип используемого алгоритма заполнения.
Необязательные атрибуты (см. Attrs
):
- resize_align_corners: если true, центры 4 угловых пикселей входного и выходного тензоров выравниваются, сохраняя значения в угловых пикселях. По умолчанию ложь.
Возврат:
-
Output
: выходной тензор.
Конструкторы и деструкторы | |
---|---|
FusedResizeAndPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input size, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) | |
FusedResizeAndPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input size, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding, const FusedResizeAndPadConv2D::Attrs & attrs) |
Публичные атрибуты | |
---|---|
operation | |
output |
Общественные функции | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Публичные статические функции | |
---|---|
ResizeAlignCorners (bool x) |
Структуры | |
---|---|
tensorflow::ops::FusedResizeAndPadConv2D::Attrs | Дополнительные установщики атрибутов для FusedResizeAndPadConv2D . |
Публичные атрибуты
операция
Operation operation
выход
::tensorflow::Output output
Общественные функции
FusedResizeAndPadConv2D
FusedResizeAndPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input size, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
FusedResizeAndPadConv2D
FusedResizeAndPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input size, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding, const FusedResizeAndPadConv2D::Attrs & attrs )
узел
::tensorflow::Node * node() const
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const
Публичные статические функции
Изменить размер, выровнять по углам
Attrs ResizeAlignCorners( bool x )