#pragma once
#include <torch/torch.h>
#include <string>

void save_tensor_to_file(const torch::Tensor& tensor, const std::string& file_path);