code: def tilt_dist(input):
      # Tilt variance by channel.
      rgb = input.transpose(0, 1)
      rgb[0] *= 1
      rgb[1] *= 10
      rgb[2] *= 100
  
      # Tilt mean by single batch.
      for i, single in enumerate(input):
          single += 2**i
  
      return input
  
file_path: fairscale/tests/nn/pipe/test_deferred_batch_norm.py
imports: from copy import deepcopy
  from itertools import chain
  import pytest
  import torch
  from torch import nn, optim
  from fairscale.nn.pipe.batchnorm import DeferredBatchNorm
label_desc: Use image node_id 1 for calling a global function with example usage: tilt_dist(input) and returns: input