bug fix in dense

release/0.11.0
peterzhang2029 8 years ago
parent 5d3e816717
commit d6bd5b1954

@ -2,6 +2,7 @@
#include <time.h>
#include <iostream>
#include <vector>
#include "../common/common.h"
#define CONFIG_BIN "./trainer_config.bin"
@ -40,7 +41,7 @@ int main() {
for (int i = 0; i < input.size(); ++i) {
input[i] = rand() / ((float)RAND_MAX);
}
// Set value for the input matrix
CHECK(paddle_matrix_set_value(mat, input.data()));
@ -66,7 +67,7 @@ int main() {
printf("Prob: \n");
for (int i = 0; i < height * width; ++i) {
printf("%.4f ", result[i]);
if ((i + 1) % width == 0){
if ((i + 1) % width == 0) {
printf("\n");
}
}

Loading…
Cancel
Save