Runtime check of asked data in tensor::get_data serizba#91

Author: serizba

include/cppflow/tensor.h

@@ -215,6 +215,16 @@ namespace cppflow {
 
     template<typename T>
     std::vector<T> tensor::get_data() const {
+
+        // Check if asked datatype and tensor datatype match
+        if (this->dtype() != deduce_tf_type<T>()) {
+            auto type1 = cppflow::to_string(deduce_tf_type<T>());
+            auto type2 = cppflow::to_string(this->dtype());
+            auto error = "Datatype in function get_data (" + type1 + ") does not match tensor datatype (" + type2 + ")";
+            throw std::runtime_error(error);
+        }
+
+
         auto res_tensor = get_tensor();
 
         // Check tensor data is not empty