tf.compat.v1.nn.separable_conv2d

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

tf.compat.v1.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.