Function TEST_F

tests/test_tensor_masking.cpp:84–105

// ============= Comparison Operations Tests =============

TEST_F(TensorMaskingTest, ComparisonEqual) {
    std::vector<float> a_data = {1, 2, 3, 4, 5};
    std::vector<float> b_data = {1, 0, 3, 0, 5};

    auto a_custom = Tensor::from_vector(a_data, {5}, Device::CUDA);
    auto b_custom = Tensor::from_vector(b_data, {5}, Device::CUDA);
    auto t = Tensor::zeros({1, 2});

    auto a_torch = torch::tensor(a_data, torch::kCUDA);
    auto b_torch = torch::tensor(b_data, torch::kCUDA);

    auto mask_custom = a_custom.eq(b_custom);
    auto mask_torch = a_torch.eq(b_torch);

    compare_tensors(mask_custom, mask_torch, 1e-5f, 1e-7f, "ComparisonEqual");

    // Test scalar comparison
    auto mask2_custom = a_custom.eq(3.0f);
    auto mask2_torch = a_torch.eq(3.0f);

    compare_tensors(mask2_custom, mask2_torch, 1e-5f, 1e-7f, "ComparisonEqualScalar");
}

TEST_F(TensorMaskingTest, ComparisonLessThan) {
    std::vector<float> a_data = {1, 2, 3, 4, 5};