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