test/custom_operator/op.h

#include <torch/op.h>

#include <cstddef>
#include <vector>

TORCH_API std::vector<at::Tensor> custom_op(
    at::Tensor tensor,
    double scalar,
    int64_t repeat);