Module flowcon.CNF.neural_odes.squeeze
Functions
def squeeze(input, downscale_factor=2)-
[:, C, Hr, Wr] -> [:, C*r^2, H, W]
def unsqueeze(input, upscale_factor=2)-
[:, Cr^2, H, W] -> [:, C, Hr, W*r]
flowcon.CNF.neural_odes.squeeze
def squeeze(input, downscale_factor=2)
[:, C, Hr, Wr] -> [:, C*r^2, H, W]
def unsqueeze(input, upscale_factor=2)
[:, Cr^2, H, W] -> [:, C, Hr, W*r]