class_docstr: Test functions defined in `functional` module
class_docstr_tok: ["Test","functions","defined","in","`","functional","`","module"]
code: def test_griffinlim(self):
      n_fft = 400
      ws = 400
      hop = 200
      window = torch.hann_window(ws)
      power = 2
      momentum = 0.99
      n_iter = 32
      length = 1000
      batch = torch.rand(self.batch_size, 1, 201, 6)
      kwargs = {
          "window": window,
          "n_fft": n_fft,
          "hop_length": hop,
          "win_length": ws,
          "power": power,
          "n_iter": n_iter,
          "momentum": momentum,
          "length": length,
          "rand_init": False,
      }
      func = partial(F.griffinlim, **kwargs)
      self.assert_batch_consistency(func, inputs=(batch,), atol=1e-4)
  
imports: import itertools
  import math
  from functools import partial
  import torch
  import torchaudio.functional
  from parameterized import parameterized, parameterized_class
  from torchaudio_unittest import common_utils
label_desc: Use image node_id 2 for calling the TestFunctional obj's underlying member method code with example usage: obj.test_griffinlim() without return types