tensorflow::ops::Conv2D

#include <nn_ops.h>

Вычисляет 2-мерную свёртку для тензоров 4-го порядка input и filter.

Краткое описание

При заданном входном тензоре формы [batch, in_height, in_width, in_channels] и тензоре фильтра/ядра формы [filter_height, filter_width, in_channels, out_channels], этот оператор выполняет следующее:

1. Расплющивает фильтр в 2-мерную матрицу с формой [filter_height * filter_width * in_channels, output_channels].
2. Извлекает фрагменты изображения из входного тензора, чтобы сформировать виртуальный тензор формы [batch, out_height, out_width, filter_height * filter_width * in_channels].
3. Для каждого фрагмента, выполняет правое умножение матрицы фильтра и вектора фрагмента изображения.

Подробно, с форматом NHWC по умолчанию,

output[b, i, j, k] =
    sum_{di, dj, q} input[b, strides[1] * i + di, strides[2] * j + dj, q] *
                    filter[di, dj, q, k]

Должно выполняться условие strides[0] = strides[3] = 1. Для наиболее распространённого случая одинаковых горизонтальных и вертикальных шагов, strides = [1, stride, stride, 1].

Аргументы:

• scope: Объект Scope
• input: Тензор 4-го порядка. Порядок измерения интерпретируется в соответствии со значением data_format, см. подробности ниже.
• filter: Тензор 4-го порядка формы [filter_height, filter_width, in_channels, out_channels]
• strides: 1-мерный тензор длиной 4. Шаг скользящего окна для каждой размерности input. Порядок размерностей определяется значением data_format, см. подробности ниже.
• padding: Тип алгоритма заполнения.

Необязательные атрибуты (см. Attrs):

• explicit_paddings: Если padding является "EXPLICIT", список явных значений заполнения. Для i-ой размерности, количество заполнения перед и после размерности составляет соответственно explicit_paddings[2 * i] и explicit_paddings[2 * i + 1]. Если padding не является "EXPLICIT", то explicit_paddings должно быть пустым.
• data_format: Указывает формат данных входных и выходных данных. С форматом "NHWC" по умолчанию данные хранятся в порядке: [batch, высота, ширина, каналы]. В качестве альтернативы, формат может быть "NCHW", порядок хранения данных: [batch, каналы, высота, ширина].
• dilations: 1-мерный тензор длиной 4. Коэффициент расширения для каждой размерности input. Если установлено значение k > 1, между каждым элементом фильтра в данной размерности будут пропущены k-1 ячеек. Порядок размерностей определяется значением data_format, см. выше для подробностей. Расширения по размерностям пакетной обработки и глубины должны быть равны 1.

Возвращает:

• Output: Тензор 4-го порядка. Порядок размерностей определяется значением data_format, см. ниже для подробностей.

Публичные атрибуты

Operation operation

::tensorflow::Output output

Публичные функции

 Conv2D(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input filter,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

 Conv2D(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input filter,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding,
  const Conv2D::Attrs & attrs
)

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const 

Публичные статические функции

Attrs DataFormat(
  StringPiece x
)

Attrs Dilations(
  const gtl::ArraySlice< int > & x
)

Attrs ExplicitPaddings(
  const gtl::ArraySlice< int > & x
)

Attrs UseCudnnOnGpu(
  bool x
)