тензорный поток :: ops :: FusedPadConv2D

#include <nn_ops.h>

Выполняет заполнение как предварительную обработку во время свертки.

Резюме

Подобно FusedResizeAndPadConv2d, эта операция позволяет оптимизировать реализацию, в которой этап преобразования пространственного заполнения объединяется с поиском im2col, но в этом случае без билинейной фильтрации, необходимой для изменения размера. Объединение заполнения предотвращает необходимость записывать промежуточные результаты в виде целых тензоров, уменьшая нагрузку на память, и мы можем получить некоторое увеличение задержки путем объединения вычислений преобразования. Атрибут data_format для Conv2D не поддерживается этой операцией, вместо этого используется порядок NHWC. Внутри этой операции используется один рабочий буфер для каждого графа, что означает, что он будет блокироваться, если несколько версий выполняются параллельно. Это связано с тем, что этот оператор в первую очередь оптимизирует использование памяти.

Аргументы:

  • scope: объект Scope
  • ввод: 4-D с формой [batch, in_height, in_width, in_channels] .
  • paddings: матрица из двух столбцов, определяющая размеры заполнения. Количество строк должно совпадать с рангом input .
  • фильтр: 4-D с формой [filter_height, filter_width, in_channels, out_channels] .
  • шаги: 1-D длины 4. Шаг скользящего окна для каждого измерения input . Должен быть в том же порядке, что и размер, указанный в format.
  • 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

выход

::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 

оператор :: тензорный поток :: Ввод

 operator::tensorflow::Input() const 

оператор :: tenorflow :: Выход

 operator::tensorflow::Output() const