fix: cross_section array order was incorrect

seung-lab · Mar 13, 2024 · 6827c92 · 6827c92
1 parent f175d41
commit 6827c92
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/fastxs3d.cpp b/src/fastxs3d.cpp
@@ -26,7 +26,7 @@ auto section(
 
 	const uint64_t voxels = sx * sy * sz;
 
-	py::array_t arr = py::array_t<float, py::array::f_style>(voxels);
+	py::array_t arr = py::array_t<float, py::array::f_style>({ sx, sy, sz });
     float* data = static_cast<float*>(arr.request().ptr);
     std::fill(data, data + voxels, 0.0f);
 
@@ -39,7 +39,7 @@ auto section(
 		data
 	);
 
-	return arr.reshape({ sx, sy, sz });
+	return arr;
 }
 
 auto area(