class_docstr: This class implements an NMF decoder
  
  Arguments
  ---------
  n_comp : int
      Number of NMF components
  n_freq : int
      The number of frequency bins in the NMF dictionary
  device : str
      The device to run the model
  
  Example:
  --------
  >>> NMF_dec = NMFDecoderAudio(20, 210, device='cpu')
  >>> H = torch.rand(1, 20, 150)
  >>> Xhat = NMF_dec.forward(H)
  >>> print(Xhat.shape)
  torch.Size([1, 210, 150])
class_docstr_tok: ["This","class","implements","an","NMF","decoder","Arguments","--","--","--","--","-","n_comp",":","int","Number","of","NMF","components","n_freq",":","int","The","number","of","frequency","bins","in","the","NMF","dictionary","device",":","str","The","device","to","run","the","model","Example",":","--","--","--","--",">",">",">","NMF_dec","=","NMFDecoderAudio","(","20",",","210",",","device='cpu","'",")",">",">",">","H","=","torch.rand","(","1",",","20",",","150",")",">",">",">","Xhat","=","NMF_dec.forward","(","H",")",">",">",">","print","(","Xhat.shape",")","torch.Size","(","[","1",",","210",",","150","]",")"]
docstr: This function returns the NMF dictionary
docstr_tok: ["This","function","returns","the","NMF","dictionary"]
code: def return_W(self):
      """This function returns the NMF dictionary"""
      W = self.W
      return self.activ(W)
  
imports: import torch.nn
  import torch.nn.functional
  import torch
  from speechbrain.lobes.models.PIQ import ResBlockAudio
label_desc: Use image node_id 3 for calling the NMFDecoderAudio obj's underlying member method code with example usage: obj.return_W() and returns: self