code: def __init__(self, input_dim=3, degree=4):
      super().__init__()
  
      self.input_dim = input_dim
      self.degree = degree
      self.output_dim = input_dim + input_dim * 2 * degree
  
file_path: NeMo/nemo/collections/multimodal/modules/nerf/utils/torch_ngp/freqencoder.py
imports: import _freqencoder
  import torch
  import torch.nn
  from torch.autograd import Function
  from torch.cuda.amp import custom_bwd, custom_fwd
label_desc: Use image node_id 1 to create a new FreqEncoder object from inherited base classes: nn with example: obj = FreqEncoder(input_dim, degree)