Fix output channel calculation in PixelShuffle2DUpBlock - #40

Open

wli51 wants to merge 4 commits into

WayScience:mainfrom

wli51:upsample-fix

Commits on Aug 19, 2026

Fix output channel calculation in PixelShuffle2DUpBlock to account for spatial downsample factor. Erroenous behavior from previous refactor due to lack of caution. With pixel shuffle upsampling, th…
wli51
committed
fc50d79
Add preserve_channels option to PixelShuffle2DUpBlock to support both channel perserving and unpreserving behavior for maximized backward compatibility. The default behavior for unext initializatio…
wli51
committed
44d1c1f

Commits on Aug 21, 2026