Skip to content

Commit ef59839

Browse files
committed
fix type casting error
1 parent ec348bb commit ef59839

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/tensor/main.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ void test1(const bool is_cpu) {
6767
}
6868

6969
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];
7171
if(shape == ndim) {
7272
std::cout << "tensor::get_tensor() test1-5: pass" << std::endl;
7373
} else {

0 commit comments

Comments
 (0)