The size parameter is used to specify the spatial size of the output for nontrainable mode ( see: https://pytorch.org/docs/stable/generated/torch.nn.Upsample.html). I think this parameter should be added into our current UpSample module.