Function TEST_F

tests/test_tensor_vs_torch.cpp:239–254

// ============================================================================

TEST_F(TensorVsTorchTest, BasicCreation) {
    // Test zeros
    auto torch_zeros = torch::zeros({10, 5}, torch::kCUDA);
    auto gs_zeros = lfs::core::Tensor::zeros({10, 5}, lfs::core::Device::CUDA);
    EXPECT_TRUE(tensors_equal(torch_zeros, gs_zeros));

    // Test ones
    auto torch_ones = torch::ones({10, 5}, torch::kCUDA);
    auto gs_ones = lfs::core::Tensor::ones({10, 5}, lfs::core::Device::CUDA);
    EXPECT_TRUE(tensors_equal(torch_ones, gs_ones));

    // Test full
    auto torch_full = torch::full({10, 5}, 3.14f, torch::kCUDA);
    auto gs_full = lfs::core::Tensor::full({10, 5}, 3.14f, lfs::core::Device::CUDA);
    EXPECT_TRUE(tensors_equal(torch_full, gs_full));
}

TEST_F(TensorVsTorchTest, RandInt) {
    // Test randint with same seed