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: Reset module parameters.
  
  Args:
      size: Block size.
      block_id: Block index.
      num_blocks: Number of blocks in the architecture.
docstr_tok: ["Reset","module","parameters",".","Args",":","size",":","Block","size",".","block_id",":","Block","index",".","num_blocks",":","Number","of","blocks","in","the","architecture","."]
code: def reset_parameters(
      self, size: int, block_id: int, num_blocks: int
  ) -> None:
      """Reset module parameters.
  
      Args:
          size: Block size.
          block_id: Block index.
          num_blocks: Number of blocks in the architecture.
  
      """
      ratio_1_to_almost0 = 1.0 - (block_id / num_blocks)
  
      time_weight = torch.ones(1, 1, size)
  
      for i in range(size):
          time_weight[0, 0, i] = i / size
  
      with torch.no_grad():
          self.time_mix_key.data = torch.pow(
              time_weight, ratio_1_to_almost0
          )
          self.time_mix_receptance.data = torch.pow(
              time_weight, ratio_1_to_almost0
          )
  
imports: from typing import List, Optional, Tuple
  import torch
label_desc: Use image node_id 2 for calling the FeedForward obj's underlying member method code with example usage: obj.reset_parameters(size, block_id, num_blocks) without return types