code: def _reset_sample_grads(module: Module) -> None:
      module.weight.sample_grad = 0  # type: ignore
      if module.bias is not None:
          module.bias.sample_grad = 0
  
imports: from collections import defaultdict
  from enum import Enum
  from typing import cast, DefaultDict, Iterable, List, Optional, Tuple, Union
  import torch
  from captum._utils.common import _format_tensor_into_tuples, _register_backward_hook
  from torch import Tensor
  from torch.nn import Module
label_desc: Use image node_id 1 for calling a global function with example usage: _reset_sample_grads(module) without return types