Fixing Clip in GetSubTensor

mfarag13 · essayed · commit 665585bc62d0 · 2022-10-17T11:08:44.000Z
diff --git a/include/mli_iterator.hpp b/include/mli_iterator.hpp
@@ -1003,7 +1003,7 @@ class TensorIterator {
         } else { // Middle iteration
           copysize[dim] = m_config.get_size(r);
         }
-        copysize[dim] = MIN(m_full_tensor.get_dim(dim) - m_pos[dim], copysize[dim]);
+        copysize[dim] = MIN(m_full_tensor.get_dim(dim) - pos[dim], copysize[dim]);
       }
       return m_full_tensor.slice(pos, copysize);
     }

Original file line number	Diff line number	Diff line change
`@@ -1003,7 +1003,7 @@ class TensorIterator {`
`1003`	`1003`	`} else { // Middle iteration`
`1004`	`1004`	`copysize[dim] = m_config.get_size(r);`
`1005`	`1005`	`}`
`1006`		`- copysize[dim] = MIN(m_full_tensor.get_dim(dim) - m_pos[dim], copysize[dim]);`
	`1006`	`+ copysize[dim] = MIN(m_full_tensor.get_dim(dim) - pos[dim], copysize[dim]);`
`1007`	`1007`	`}`
`1008`	`1008`	`return m_full_tensor.slice(pos, copysize);`
`1009`	`1009`	`}`