code: def forward(self, z):
      frequencies_offsets = self.network(z)
      frequencies = frequencies_offsets[
          ..., : frequencies_offsets.shape[-1] // 2
      ]
      phase_shifts = frequencies_offsets[
          ..., frequencies_offsets.shape[-1] // 2 :
      ]
  
      return frequencies, phase_shifts
  
imports: import os
  import time
  import timm
  import glob
  import numpy
  import os.path
  import torch
  import torch.distributed
  from torch import nn
  import torch.nn.functional
  from .iresnet import get_model
label_desc: Use image node_id 2 for calling the CustomMappingNetwork obj's underlying member method code with example usage: obj.forward(z) and returns: frequencies,  phase_shifts