text: import torch
  
  
  class InverseNormalize:
      def __init__(self, mean, std):
          self.mean = torch.Tensor(mean)[None, :, None, None]
          self.std = torch.Tensor(std)[None, :, None, None]
  
      def __call__(self, sample):
          return (sample * self.std) + self.mean
  
      def to(self, device):
          self.mean = self.mean.to(device)
          self.std = self.std.to(device)
          return self
  
  
  class InverseNormalizeNone:
      def __init__(self):
          pass
  
      def __call__(self, sample):
          return sample
  
      def to(self, device):
          return self
  