torch-ext/torch_binding.h

#pragma once

#include <torch/torch.h>

void mel_spectrogram(
    const torch::Tensor &out,
    const torch::Tensor &samples,
    const torch::Tensor &filters,
    const int64_t fft_size,
    const int64_t fft_step);