def forward(self, x):
    assert (x.shape[1] == self.channels)
    if (self.dims == 3):
        x = F.interpolate(x, (x.shape[2], (x.shape[3] * 2), (x.shape[4] * 2)), mode='nearest')
    else:
        x = F.interpolate(x, scale_factor=2, mode='nearest')
    if self.use_conv:
        x = self.conv(x)
    return x
def forward(self, x):
    assert (x.shape[1] == self.channels)
    if (self.dims == 3):
        x = F.interpolate(x, (x.shape[2], (x.shape[3] * 2), (x.shape[4] * 2)), mode='nearest')
    else:
        x = F.interpolate(x, scale_factor=2, mode='nearest')
    if self.use_conv:
        x = self.conv(x)
    return x