class_docstr: Upscale the frequency and time dimensions of a spectrogram.
  
  Args:
      time_scale: the scale factor in time dimension
      freq_scale: the scale factor in frequency dimension
  
  Examples
      >>> stretch2d = Stretch2d(time_scale=10, freq_scale=5)
  
      >>> input = torch.rand(10, 100, 512)  # a random spectrogram
      >>> output = stretch2d(input)  # shape: (10, 500, 5120)
class_docstr_tok: ["Upscale","the","frequency","and","time","dimensions","of","a","spectrogram",".","Args",":","time_scale",":","the","scale","factor","in","time","dimension","freq_scale",":","the","scale","factor","in","frequency","dimension","Examples",">",">",">","stretch2d","=","Stretch2d","(","time_scale=10",",","freq_scale=5",")",">",">",">","input","=","torch.rand","(","10",",","100",",","512",")","#","a","random","spectrogram",">",">",">","output","=","stretch2d","(","input",")","#","shape",":","(","10",",","500",",","5120",")"]
code: def __init__(self, time_scale: int, freq_scale: int) -> None:
      super().__init__()
  
      self.freq_scale = freq_scale
      self.time_scale = time_scale
  
imports: import math
  from typing import List, Optional, Tuple
  import torch
  import torch.nn.functional
  from torch import nn, Tensor
label_desc: Use image node_id 1 to create a new Stretch2d object from inherited base classes: nn with example: obj = Stretch2d(time_scale, freq_scale)