This is the code I used for nn.MSELoss. Does output and target shapes here refer to pred_fake and label_real in the code, respectively?I printed out their shapes.

fake_B = netG_A2B(real_A)
        pred_fake = netD_B(fake_B)
        loss_GAN_A2B = criterion_GAN(pred_fake, label_real)

criterion_GAN = torch.nn.MSELoss()


2. In other words, if CUDA is available, then I need to push the input tensor from the CPU to the GPU. If CUDA is not used, the input tensor on the CPU can directly enter the network without the need to do Variable (tensor) processing, because the effect is the same, right?
PS: In addition to the code snippets, should the result of the operation also be wrapped into three backticks ```? To replace those pictures?