Function TEST_F

tests/test_tensor_reduction.cpp:80–91

// ============= Basic Scalar Reductions =============

TEST_F(TensorReductionTest, Sum) {
    std::vector<float> data = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};

    auto custom_tensor = Tensor::from_vector(data, {10}, Device::CUDA);
    auto torch_tensor = torch::tensor(data, torch::TensorOptions().device(torch::kCUDA));

    float custom_sum = custom_tensor.sum_scalar();
    float torch_sum = torch_tensor.sum().item<float>();

    compare_scalars(custom_sum, torch_sum, 1e-4f, "Sum");
    EXPECT_FLOAT_EQ(custom_sum, 55.0f);
}

TEST_F(TensorReductionTest, SumMultiDim) {
    std::vector<float> data(24);