class_docstr: Helper class to simplify the handling of parameter buckets
class_docstr_tok: ["Helper","class","to","simplify","the","handling","of","parameter","buckets"]
code: def __init__(
      self, size: int, dtype: torch.dtype, device: torch.device
  ) -> None:
      super().__init__(size, dtype, device)
  
imports: from typing import Any, Callable, List, Optional, Union
  import torch
label_desc: Use image node_id 1 to create a new ParamBucket object from inherited base classes: Bucket with example: obj = ParamBucket(size, dtype, device)