old_contents: # AUTOGENERATED! DO NOT EDIT! File to edit: nbs/99_pytorch_doc.ipynb (unless otherwise specified).
  
  __all__ = ['PYTORCH_URL', 'pytorch_doc_link']
  
  # Cell
  from types import ModuleType
  
  # Cell
  PYTORCH_URL = 'https://pytorch.org/docs/stable/'
  
  # Cell
  def _mod2page(
      mod:ModuleType, # A pytorch module
  ) -> str:
      "Get the web page name for a pytorch module"
      if mod == Tensor: return 'tensors.html'
      name = mod.__name__
      name = name.replace('torch.', '').replace('utils.', '')
      if name.startswith('nn.modules'): return 'nn.html'
      return f'{name}.html'
  
  # Cell
  import importlib
  
  # Cell
  def pytorch_doc_link(
      name:str, # Name of a pytorch module, class or function
  ) -> (str, None):
      "Get the URL to the documentation of a pytorch module, class or function"
      if name.startswith('F'): name = 'torch.nn.functional' + name[1:]
      if not name.startswith('torch.'): name = 'torch.' + name
      if name == 'torch.Tensor': return f'{PYTORCH_URL}tensors.html'
      try:
          mod = importlib.import_module(name)
          return f'{PYTORCH_URL}{_mod2page(mod)}'
      except: pass
      splits = name.split('.')
      mod_name,fname = '.'.join(splits[:-1]),splits[-1]
      if mod_name == 'torch.Tensor': return f'{PYTORCH_URL}tensors.html#{name}'
      try:
          mod = importlib.import_module(mod_name)
          page = _mod2page(mod)
          return f'{PYTORCH_URL}{page}#{name}'
      except: return None
new_contents: # AUTOGENERATED! DO NOT EDIT! File to edit: nbs/99_pytorch_doc.ipynb (unless otherwise specified).
  
  __all__ = ['PYTORCH_URL', 'pytorch_doc_link']
  
  # Cell
  from types import ModuleType
  
  # Cell
  PYTORCH_URL = 'https://pytorch.org/docs/stable/'
  
  # Cell
  def _mod2page(
      mod:ModuleType, # A PyTorch module
  ) -> str:
      "Get the web page name for a pytorch module"
      if mod == Tensor: return 'tensors.html'
      name = mod.__name__
      name = name.replace('torch.', '').replace('utils.', '')
      if name.startswith('nn.modules'): return 'nn.html'
      return f'{name}.html'
  
  # Cell
  import importlib
  
  # Cell
  def pytorch_doc_link(
      name:str, # Name of a pytorch module, class or function
  ) -> (str, None):
      "Get the URL to the documentation of a pytorch module, class or function"
      if name.startswith('F'): name = 'torch.nn.functional' + name[1:]
      if not name.startswith('torch.'): name = 'torch.' + name
      if name == 'torch.Tensor': return f'{PYTORCH_URL}tensors.html'
      try:
          mod = importlib.import_module(name)
          return f'{PYTORCH_URL}{_mod2page(mod)}'
      except: pass
      splits = name.split('.')
      mod_name,fname = '.'.join(splits[:-1]),splits[-1]
      if mod_name == 'torch.Tensor': return f'{PYTORCH_URL}tensors.html#{name}'
      try:
          mod = importlib.import_module(mod_name)
          page = _mod2page(mod)
          return f'{PYTORCH_URL}{page}#{name}'
      except: return None
subject: Update fastai/_pytorch_doc.py