From 5f48d786ead5df60ead0bcd7390354826ec6757e Mon Sep 17 00:00:00 2001 From: a1den Date: Sat, 5 Jul 2025 12:41:41 +0700 Subject: [PATCH] Implement fast_sigmoid C++ core and Python binding --- extension_cpp/csrc/include/extension/fast_sigmoid.h | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 extension_cpp/csrc/include/extension/fast_sigmoid.h diff --git a/extension_cpp/csrc/include/extension/fast_sigmoid.h b/extension_cpp/csrc/include/extension/fast_sigmoid.h new file mode 100644 index 0000000..c92e6bd --- /dev/null +++ b/extension_cpp/csrc/include/extension/fast_sigmoid.h @@ -0,0 +1,13 @@ +#pragma once +#include + +namespace extension_cpp { + +TORCH_API torch::Tensor fast_sigmoid( + const torch::Tensor& input, + double min = -10.0, + double max = 10.0, + int64_t num_entries = 1000 +); + +} // namespace extension_cpp