tf.nn.separable_conv2d

Двухмерная свёртка с раздельными фильтрами.

tf.nn.separable_conv2d(
    input,
    depthwise_filter,
    pointwise_filter,
    strides,
    padding,
    data_format=None,
    dilations=None,
    name=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.