PhaneeshB

import torch

## Example Inputs 
num_mel_bins=torch.as_tensor([8], dtype=torch.int)
dft_length=torch.as_tensor([16], dtype=torch.int)
sample_rate=torch.as_tensor([8192], dtype=torch.int)
lower_edge_hertz=torch.Tensor([0.0])
upper_edge_hertz=torch.Tensor([8192/2])

PhaneeshB

import torch

num_mel_bins = torch.as_tensor([8], dtype=torch.int)
dft_length = torch.as_tensor([16], dtype=torch.int)
sample_rate = torch.as_tensor([8192], dtype=torch.int)
lower_edge_hertz = torch.Tensor([0.0])
upper_edge_hertz = torch.Tensor([8192/2])
output_datatype = None

num_spectrogram_bins = dft_length // 2 + 1