layers.py 文件源码

def nn_upsample(upsample_in, num_styles=None, num_filters=None, filter_size=3, stride=1):
    if num_filters == None:
        num_filters = upsample_in.output_shape[1]

    nn_network = ExpressionLayer(upsample_in, lambda X: X.repeat(2, 2).repeat(2, 3), output_shape='auto')
    nn_network = style_conv_block(nn_network, num_styles, num_filters, filter_size, stride)

    return nn_network