code: def test_istft(device):
      from speechbrain.processing.features import STFT
      from speechbrain.processing.features import ISTFT
  
      fs = 16000
      inp = torch.randn([10, 16000], device=device)
      inp = torch.stack(3 * [inp], -1)
  
      compute_stft = STFT(sample_rate=fs).to(device)
      compute_istft = ISTFT(sample_rate=fs).to(device)
      out = compute_istft(compute_stft(inp), sig_length=16000)
  
      assert torch.sum(torch.abs(inp - out) < 5e-5) >= inp.numel() - 5
  
      assert torch.jit.trace(compute_stft, inp)
      assert torch.jit.trace(compute_istft, compute_stft(inp))
  
imports: import torch
label_desc: Use image node_id 3 for calling a global function with example usage: test_istft(device) without return types