提交 d6bd5b19 编写于 作者: P peterzhang2029

bug fix in dense

上级 5d3e8167
......@@ -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");
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册