distance_gan_model.py 文件源码

def get_self_distances(self):

        A_half_1, A_half_2 = torch.chunk(self.real_A, 2, dim=2)
        B_half_1, B_half_2 = torch.chunk(self.real_B, 2, dim=2)
        AB_half_1, AB_half_2 = torch.chunk(self.fake_B, 2, dim=2)
        BA_half_1, BA_half_2 = torch.chunk(self.fake_A, 2, dim=2)

        l_distance_A, l_distance_B = \
            self.get_individual_distance_loss(A_half_1, A_half_2,
                                              AB_half_1, AB_half_2,
                                              B_half_1, B_half_2,
                                              BA_half_1, BA_half_2)

        return l_distance_A, l_distance_B