Apply dilation rate individually to each dimension with tf.nn.atrous_conv2d?

Question

I have a tensor with shape B x H x W x C and I like to apply dilation to only H . Do you know any way to have this small trick with tf.nn.atrous_conv2d ?

Answer 1

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