#ifndef UTILS_H
#define UTILS_H

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

namespace TyTorchUtils {
  // Parse dtype string to torch::ScalarType
  torch::ScalarType ParseDtype(const std::string& dtype_str);

  // Parse device string to torch::Device
  torch::Device ParseDevice(const std::string& device_str);
}

#endif