Function TEST_F

tests/test_tensor_ops.cpp:103–122

// ============= Scalar Operations =============

TEST_F(TensorOpsTest, ScalarAdd) {
    std::vector<float> data(12);
    for (auto& val : data)
        val = dist(gen);

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

    float scalar = 2.5f;

    auto custom_result = custom_tensor.add(scalar);
    auto torch_result = torch_tensor + scalar;

    compare_tensors(custom_result, torch_result, 1e-5f, 1e-6f, "ScalarAdd");

    // Test operator overload
    auto custom_result2 = custom_tensor + scalar;
    compare_tensors(custom_result2, torch_result, 1e-5f, 1e-6f, "ScalarAdd_Operator");
}

TEST_F(TensorOpsTest, ScalarSubtract) {
    std::vector<float> data(10);