visualizer.py 文件源码

def make_code(self, num_rows):
        z_dim = [int(np.prod(self.network.code_dims))]
        code_x = torch.linspace(-2, 2, steps=num_rows).view(1, num_rows).repeat(num_rows, 1)
        code_y = code_x.t()
        if self.args.ngpus > 0:
            z = torch.cuda.FloatTensor(*z_dim).normal_()
            code = torch.stack([code_x, code_y], dim=2).view(-1,2).cuda()
        else:
            z = torch.FloatTensor(*z_dim).normal_()
            code = torch.stack([code_x, code_y], dim=2).view(-1,2)
        return code