class_docstr: Module for partial gradient computation.
  Useful when computing loss on batch splits to save memory.
class_docstr_tok: ["Module","for","partial","gradient","computation",".","Useful","when","computing","loss","on","batch","splits","to","save","memory","."]
code: def __init__(self, func: torch.nn.Module):
      super().__init__()
      self.func = func
  
file_path: NeMo/nemo/collections/asr/parts/k2/grad_utils.py
imports: import torch
  from nemo.collections.asr.parts.k2.utils import make_non_pad_mask
label_desc: Use image node_id 1 to create a new PartialGrad object from inherited base classes: torch.nn with example: obj = PartialGrad(func)