for (int i = 0; i < output_size; i++) {
            output[i] = 0;
            for (int j = 0; j < input_size; j++) {
                output[i] += input[j] * weights[j][i];
            }