test_ops.h

#pragma once
#include <torch/library.h>
#include <ATen/ATen.h>

// 声明操作符实现函数
namespace test_ops_impl {
    at::Tensor add_one(at::Tensor input);
    at::Tensor multiply_by_two(at::Tensor input);
}