Fix: make sure background and alpha are on the same device

[jb-ye]

Thanks, closed as dup of https://github.com/nerfstudio-project/nerfstudio/pull/3219