code: def reset_parameters(self) -> None:
      with torch.no_grad():
          for key, _length in self.max_feature_lengths.items():
              self.position_weights[key].fill_(1.0)
              # Re-assign python dict to param dict in case of re-materialization
              self.position_weights_dict[key] = self.position_weights[
                  key
              ]
  
imports: import abc
  from typing import Dict, Optional
  import torch
  from torch import nn
  from torchrec.sparse.jagged_tensor import JaggedTensor, KeyedJaggedTensor
label_desc: Use image node_id 2 for calling the PositionWeightedModuleCollection obj's underlying member method code with example usage: obj.reset_parameters() without return types