Commit 83faec77 authored by rusty1s's avatar rusty1s
Browse files

fix CPP build

parent c2350bb6
...@@ -17,17 +17,17 @@ ...@@ -17,17 +17,17 @@
}() }()
template <typename scalar_t> template <typename scalar_t>
torch::Tensor from_vector(const std::vector<scalar_t> &vec, inline torch::Tensor from_vector(const std::vector<scalar_t> &vec,
bool inplace = false) { bool inplace = false) {
const auto size = (int64_t)vec.size(); const auto size = (int64_t)vec.size();
const auto out = torch::from_blob((scalar_t *)vec.data(), {size}, const auto out = torch::from_blob((scalar_t *)vec.data(), {size},
c10::CppTypeToScalarType<scalar_t>::value); c10::CppTypeToScalarType<scalar_t>::value);
return inplace ? out : out.clone(); return inplace ? out : out.clone();
} }
torch::Tensor choice(int64_t population, int64_t num_samples, inline torch::Tensor
bool replace = false, choice(int64_t population, int64_t num_samples, bool replace = false,
torch::optional<torch::Tensor> weight = torch::nullopt) { torch::optional<torch::Tensor> weight = torch::nullopt) {
if (!replace && num_samples >= population) if (!replace && num_samples >= population)
return torch::arange(population, at::kLong); return torch::arange(population, at::kLong);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment