Merge pull request serizba#111 from seungtaek94/master

serizba · web-flow · commit 9ea9519c66d9 · 2021-03-30T09:17:24.000+02:00
Fix type casting error on example code
diff --git a/examples/tensor/main.cpp b/examples/tensor/main.cpp
@@ -67,7 +67,7 @@ void test1(const bool is_cpu) {
     }
 
     auto shape_tensor = t1.shape();
-    auto shape = shape_tensor.get_data<int32_t>()[0];
+    auto shape = shape_tensor.get_data<int64_t>()[0];
     if(shape == ndim) {
         std::cout << "tensor::get_tensor() test1-5: pass" << std::endl;
     } else {

Original file line number	Diff line number	Diff line change
`@@ -67,7 +67,7 @@ void test1(const bool is_cpu) {`
`67`	`67`	`}`
`68`	`68`
`69`	`69`	`auto shape_tensor = t1.shape();`
`70`		`- auto shape = shape_tensor.get_data<int32_t>()[0];`
	`70`	`+ auto shape = shape_tensor.get_data<int64_t>()[0];`
`71`	`71`	`if(shape == ndim) {`
`72`	`72`	`std::cout << "tensor::get_tensor() test1-5: pass" << std::endl;`
`73`	`73`	`} else {`