code: def forward(self, x, label):
      x = flow.nn.functional.l2_normalize(input=x, dim=1, epsilon=1e-10)
      if self.partial_fc:
          (
              mapped_label,
              sampled_label,
              sampled_weight,
          ) = flow.distributed_partial_fc_sample(
              weight=self.weight,
              label=label,
              num_sample=self.total_num_sample,
          )
          label = mapped_label
          weight = sampled_weight
      else:
          weight = self.weight
      weight = flow.nn.functional.l2_normalize(
          input=weight, dim=1, epsilon=1e-10
      )
      x = flow.matmul(x, weight, transpose_b=True)
      if x.is_consistent:
          return x, label
      else:
          return x
  
imports: import oneflow
  from oneflow.nn.parallel import DistributedDataParallel
  from utils.ofrecord_data_utils import OFRecordDataLoader, SyntheticDataLoader
  from utils.utils_logging import AverageMeter
  from utils.utils_callbacks import CallBackVerification, CallBackLogging, CallBackModelCheckpoint
  from backbones import get_model
  from graph import TrainGraph, EvalGraph
  from utils.losses import CrossEntropyLoss_sbp
  import logging
label_desc: Use image node_id 2 for calling the FC7 obj's underlying member method code with example usage: obj.forward(x, label) and returns: x,  label, x