tf.compat.v2.nn.separable_conv2d

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

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