ext-torch/torch_binding.h

#pragma once

#include <torch/torch.h>

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