#include "sigmoid.h"
#include "../tensor.h"
#include <torch/torch.h>

namespace TensorOps {

Napi::Value Sigmoid(Tensor* self, const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();

  try {
    // Apply sigmoid activation: σ(x) = 1 / (1 + exp(-x))
    torch::Tensor result = torch::sigmoid(self->tensor);
    return Tensor::NewInstance(env, result);
  } catch (const std::exception& e) {
    Napi::Error::New(env, e.what()).ThrowAsJavaScriptException();
    return env.Undefined();
  }
}

} // namespace TensorOps