code: def __init__(
      self,
      block: Type[Bottleneck2D],
      num_blocks: int,
      planes: int,
      depth: int,
      expansion: int = 2,
  ) -> None:
      super().__init__()
      self.depth = depth
      self.block = block
      self.expansion = expansion
      self.hg = self._make_hour_glass(block, num_blocks, planes, depth)
  
imports: from typing import Any, Dict, List, NamedTuple, Optional, Tuple, Type, Union
  import torch
  import torch.nn.functional
  from torch import nn
  from torch.nn.functional import pixel_shuffle, softmax
  from kornia.core import Module, Tensor
label_desc: Use image node_id 1 to create a new Hourglass object from inherited base classes: Module with example: obj = Hourglass(block, num_blocks, planes, depth, expansion)