标签: tensorflow
似乎tf.nn.depthwise_conv2d(https://www.tensorflow.org/api_docs/python/tf/nn/depthwise_conv2d)应该是解决方案
tf.nn.depthwise_conv2d