diff --git a/‎src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs‎
Lines changed: 8 additions & 16 deletions b/‎src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs‎
Lines changed: 8 additions & 16 deletions
diff --git a/‎src/TensorFlowNET.Core/Train/Saving/Saver.cs‎
Lines changed: 2 additions & 2 deletions b/‎src/TensorFlowNET.Core/Train/Saving/Saver.cs‎
Lines changed: 2 additions & 2 deletions
@@ -36,6 +36,10 @@ private IntPtr Allocate(NDArray nd)
                 size = (ulong)(nd.size * nd.dtypesize);
             }
 
+            var dataType = ToTFDataType(nd.dtype);
+            // shape
+            var dims = nd.shape.Select(x => (long)x).ToArray();
+
             switch (nd.dtype.Name)
             {
                 case "Int16":
@@ -51,37 +55,25 @@ private IntPtr Allocate(NDArray nd)
                     Marshal.Copy(nd.Data<double>(), 0, dotHandle, nd.size);
                     break;
                 case "String":
-                    /*var value = nd.Data<string>()[0];
-                    var bytes = Encoding.UTF8.GetBytes(value);
-                    dotHandle = Marshal.AllocHGlobal(bytes.Length + 1);
-                    Marshal.Copy(bytes, 0, dotHandle, bytes.Length);
-                    size = (ulong)bytes.Length;*/
-
                     var str = nd.Data<string>()[0];
                     ulong dst_len = c_api.TF_StringEncodedSize((ulong)str.Length);
-                    //dotHandle = Marshal.AllocHGlobal((int)dst_len);
-                    //size = c_api.TF_StringEncode(str, (ulong)str.Length, dotHandle, dst_len, status);
-
                     var dataType1 = ToTFDataType(nd.dtype);
                     // shape
                     var dims1 = nd.shape.Select(x => (long)x).ToArray();
 
                     var tfHandle1 = c_api.TF_AllocateTensor(dataType1,
                         dims1,
                         nd.ndim,
-                        dst_len);
+                        dst_len + sizeof(Int64));
 
                     dotHandle = c_api.TF_TensorData(tfHandle1);
-                    c_api.TF_StringEncode(str, (ulong)str.Length, dotHandle, dst_len, status);
+                    Marshal.WriteInt64(dotHandle, 0);
+                    c_api.TF_StringEncode(str, (ulong)str.Length, dotHandle + sizeof(Int64), dst_len, status);
                     return tfHandle1;
-                    break;
                 default:
                     throw new NotImplementedException("Marshal.Copy failed.");
             }
-
-            var dataType = ToTFDataType(nd.dtype);
-            // shape
-            var dims = nd.shape.Select(x => (long)x).ToArray();
+            
             // Free the original buffer and set flag
             Deallocator deallocator = (IntPtr values, IntPtr len, ref bool closure) =>
             {
 
@@ -162,12 +162,12 @@ public string save(Session sess,
 
             if (!_is_empty)
             {
-                var model_checkpoint_path1 = sess.run(_saver_def.SaveTensorName, new FeedItem[] {
+                model_checkpoint_path = sess.run(_saver_def.SaveTensorName, new FeedItem[] {
                     new FeedItem(_saver_def.FilenameTensorName, checkpoint_file)
                 });
             }
 
-            throw new NotImplementedException("");
+            throw new NotImplementedException("Saver.save");
 
             return model_checkpoint_path;
         }
Original file line number	Diff line number	Diff line change
@@ -162,12 +162,12 @@ public string save(Session sess,
162	162
163	163	if (!_is_empty)
164	164	{
165		- var model_checkpoint_path1 = sess.run(_saver_def.SaveTensorName, new FeedItem[] {
	165	+ model_checkpoint_path = sess.run(_saver_def.SaveTensorName, new FeedItem[] {
166	166	new FeedItem(_saver_def.FilenameTensorName, checkpoint_file)
167	167	});
168	168	}
169	169
170		- throw new NotImplementedException("");
	170	+ throw new NotImplementedException("Saver.save");
171	171
172	172	return model_checkpoint_path;
173	173	}