def subpixelConv2D(*args, **kwargs):
kwargs['num_outputs'] = 4*kwargs['num_outputs']
output = tcl.conv2d(*args, **kwargs)
output = tf.depth_to_space(output, 2)
return output
微信
公众号
扫码关注公众号