Merge pull request serizba#1 from CarlPoirier/master

add tensor shape getter

Author: aul12

src/Tensor.cpp

@@ -147,7 +147,9 @@ std::vector<T> Tensor::get_data() {
     return std::vector<T>(T_data, T_data + size);
 }
 
-
+std::vector<int64_t> Tensor::get_shape() {
+	return shape;
+}
 
 template<typename T>
 TF_DataType Tensor::deduce_type() {

src/Tensor.h

@@ -40,6 +40,8 @@ class Tensor {
     template<typename T>
     std::vector<T> get_data();
 
+	std::vector<int64_t> get_shape();
+
 private:
     TF_Tensor* val;
     TF_Output op;