Method cpu

LDPS_Graph/layers/utils.py:317–319

(self)

        self.std = self.std.cuda()

    def cpu(self):
        self.mean = self.mean.cpu()
        self.std = self.std.cpu()


# normalization, scaling by range