class_docstr: FeedForward module definition.
  
  Args:
      size: Input/Output size.
      hidden_size: Hidden size.
      block_id: Block index.
      num_blocks: Number of blocks in the architecture.
docstr: Construct a FeedForward object.
docstr_tok: ["Construct","a","FeedForward","object","."]
code: def __init__(
      self, size: int, hidden_size: int, block_id: int, num_blocks: int
  ) -> None:
      """Construct a FeedForward object."""
      super().__init__()
  
      self.time_shift = torch.nn.ZeroPad2d((0, 0, 1, -1))
  
      self.time_mix_key = torch.nn.Parameter(torch.empty(1, 1, size))
      self.time_mix_receptance = torch.nn.Parameter(
          torch.empty(1, 1, size)
      )
  
      self.proj_key = torch.nn.Linear(size, hidden_size, bias=True)
      self.proj_value = torch.nn.Linear(hidden_size, size, bias=True)
      self.proj_receptance = torch.nn.Linear(size, size, bias=True)
  
      self.block_id = block_id
  
      self.reset_parameters(size, block_id, num_blocks)
  
imports: from typing import List, Optional, Tuple
  import torch
label_desc: Use image node_id 1 to create a new FeedForward object from inherited base classes: torch.nn with example: obj = FeedForward(size, hidden_size, block_id, num_blocks)