About - initial_images.cuda(), # of shape [b n c h w]

Created by: ButoneDream

hello ， could u tell more detail about ‘initial_images.cuda()’ ？