使用tf.nn.atrous_conv2d单独对每个维度应用扩张率？

Question

我有一个形状为B x H x W x C的张量，我喜欢仅将扩张应用于H。你知道如何通过tf.nn.atrous_conv2d获得这个小技巧吗？

Answer 1

tf.nn.atrous_conv2d中没有此类功能，但您可以使用tf.layers.conv2d并设置dilation_rate=(2, 1)来达到同样的效果。