[tensor] Set stride for shared tensor

Set stride for shared tensor

**Self evaluation:**
1. Build test: [x]Passed [ ]Failed [ ]Skipped
2. Run test: [x]Passed [ ]Failed [ ]Skipped

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>

diff --git a/nntrainer/tensor/tensor.cpp b/nntrainer/tensor/tensor.cpp
index da0debe6b3bd126476f2c5311a66cbf9ccc4eb44..f9e78f4a52b3839656e82a7b51220b63ab1c941f 100644 (file)
--- a/nntrainer/tensor/tensor.cpp
+++ b/nntrainer/tensor/tensor.cpp
@@ -281,6 +281,7 @@ Tensor Tensor::getSharedDataTensor(const TensorDim dim_,
 
   ret.dim = dim_;
   ret.data = std::shared_ptr<float>(this->data, this->data.get() + offset);
+  ret.strides = ret.dim.computeStrides();
 
   return ret;
 }