Fix tensor loop iteration in examples (fixes #57)

wjakob · Aug 23, 2022 · b90a6c5 · b90a6c5
1 parent 7842f2b
commit b90a6c5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/tensor.md b/docs/tensor.md
@@ -73,7 +73,7 @@ stored contiguously in CPU memory using a C-style array ordering convention.
 m.def("process", [](nb::tensor<uint8_t, nb::shape<nb::any, nb::any, 3>, nb::c_contig, nb::device::cpu> tensor) {
     // Double brightness of the MxNx3 RGB image
     for (size_t y = 0; y < tensor.shape(0); ++y)
-        for (size_t x = 0; y < tensor.shape(1); ++x)
+        for (size_t x = 0; x < tensor.shape(1); ++x)
             for (size_t ch = 0; ch < 3; ++ch)
                 tensor(y, x, ch) = (uint8_t) std::min(255, tensor(y, x, ch) * 2);
 

diff --git a/tests/test_tensor.cpp b/tests/test_tensor.cpp
@@ -80,7 +80,7 @@ NB_MODULE(test_tensor_ext, m) {
                                    nb::c_contig, nb::device::cpu> tensor) {
         // Double brightness of the MxNx3 RGB image
         for (size_t y = 0; y < tensor.shape(0); ++y)
-            for (size_t x = 0; y < tensor.shape(1); ++x)
+            for (size_t x = 0; x < tensor.shape(1); ++x)
                 for (size_t ch = 0; ch < 3; ++ch)
                     tensor(y, x, ch) = (uint8_t) std::min(255, tensor(y, x, ch) * 2);