// Function to compute Y = W * X + B
void LinearLayer(float W[], float X[], float B[], float Y[], int n)
{
    for (int i = 0; i < n; i++)
        Y[i] = W[i] * X[i] + B[i];
}