tensorflow::ops::DepthwiseConv2dNative

#include <nn_ops.h>

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

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

Исходный тензор имеет форму [batch, in_height, in_width, in_channels], а тензор фильтра/ядра имеет форму [filter_height, filter_width, in_channels, channel_multiplier], содержащий in_channels свёрточных фильтров глубиной 1. depthwise_conv2d применяет разный фильтр к каждому входному каналу (расширяя от 1 канала до channel_multiplier каналов для каждого), а затем конкатенирует результаты вместе. Таким образом, выходной тензор имеет in_channels * channel_multiplier каналов.

for k in 0..in_channels-1
  for q in 0..channel_multiplier-1
    output[b, i, j, k * channel_multiplier + q] =
      sum_{di, dj} input[b, strides[1] * i + di, strides[2] * j + dj, k] *
                        filter[di, dj, k, q]

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

Аргументы:

• scope: Объект Scope
• strides: 1-мерный массив длиной 4. Шаг скользящего окна для каждого измерения input.
• padding: Тип алгоритма заполнения.

Дополнительные атрибуты (см. Attrs):

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

Возвращаемые значения:

• Output: Выходной тензор.

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

Operation operation

::tensorflow::Output output

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

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

 DepthwiseConv2dNative(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input filter,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding,
  const DepthwiseConv2dNative::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
)