Yeah, but the Python for it would be more like:

  import sys
  import torch
  sys.modules['scratch'] = torch
  from scratch import nn