Fix stride issue for ZeroDims

riverlijunjie · Oct 25, 2023 · bbd48ca · bbd48ca
1 parent dc4240b
commit bbd48ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/inference/src/dev/make_tensor.cpp b/src/inference/src/dev/make_tensor.cpp
@@ -77,7 +77,7 @@ class ViewTensor : public ITensor {
         auto& shape = get_shape();
         if (m_strides.empty() && !shape.empty()) {
             m_strides.resize(shape.size());
-            m_strides.back() = m_element_type.size();
+            m_strides.back() = shape.back() == 0 ? 0 : m_element_type.size();
             std::transform(shape.crbegin(),
                            shape.crend() - 1,
                            m_strides.rbegin(),