tf.nn.separable_conv2d

2-D свёртка с раздельными фильтрами.

tf.nn.separable_conv2d(
    input, depthwise_filter, pointwise_filter, strides, padding, rate=None,
    name=None, data_format=None, dilations=None
)

Выполняет свёртку по глубине, которая действует раздельно на каналы, за которой следует свёртка по точкам, смешивающая каналы. Обратите внимание, что это разделимость по измерениям [1, 2] и 3, а не пространственная разделимость по измерениям 1 и 2.

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

output[b, i, j, k] = sum_{di, dj, q, r}
    input[b, strides[1] * i + di, strides[2] * j + dj, q] *
    depthwise_filter[di, dj, q, r] *
    pointwise_filter[0, 0, q * channel_multiplier + r, k]

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