Skip to content

Commit 6eeb4e1

Browse files
committed
compiled.
1 parent 918f79e commit 6eeb4e1

4 files changed

Lines changed: 13 additions & 12 deletions

File tree

src/TensorFlowNET.Core/TensorFlowNET.Core.csproj

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ Docs: https://tensorflownet.readthedocs.io</Description>
3535
13. Fix default graph and operation issue when import model.
3636
14. Fix TF_String endcode and decode.
3737
15. Fix Tensor memory leak.</PackageReleaseNotes>
38-
<LangVersion>7.2</LangVersion>
38+
<LangVersion>7.3</LangVersion>
3939
<FileVersion>0.10.7.2</FileVersion>
4040
<PackageLicenseFile>LICENSE</PackageLicenseFile>
4141
<PackageRequireLicenseAcceptance>true</PackageRequireLicenseAcceptance>

src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -486,7 +486,7 @@ public unsafe Tensor(NDArray nd, TF_DataType? tensorDType = null)
486486
{
487487
if (tensorDType == TF_DataType.TF_STRING && nd.dtype.Name == "Byte")
488488
{
489-
var buffer = nd.Data<byte>();
489+
var buffer = nd.ToArray<byte>();
490490
var size = c_api.TF_StringEncodedSize((UIntPtr)buffer.Length);
491491
var handle = TF_AllocateTensor(TF_DataType.TF_STRING, IntPtr.Zero, 0, (UIntPtr)((ulong)size + 8));
492492

@@ -524,29 +524,30 @@ private unsafe IntPtr Allocate(NDArray nd, TF_DataType? tensorDType = null)
524524
switch (nd.dtype.Name)
525525
{
526526
case "Boolean":
527-
var boolVals = Array.ConvertAll(nd1.Data<bool>(), x => Convert.ToByte(x));
527+
var boolVals = Array.ConvertAll(nd1.ToArray<bool>(), x => Convert.ToByte(x));
528528
Marshal.Copy(boolVals, 0, dotHandle, nd.size);
529529
break;
530530
case "Int16":
531-
Marshal.Copy(nd1.Data<short>(), 0, dotHandle, nd.size);
531+
Marshal.Copy(nd1.ToArray<short>(), 0, dotHandle, nd.size);
532532
break;
533533
case "Int32":
534-
Marshal.Copy(nd1.Data<int>(), 0, dotHandle, nd.size);
534+
Marshal.Copy(nd1.ToArray<int>(), 0, dotHandle, nd.size);
535535
break;
536536
case "Int64":
537-
Marshal.Copy(nd1.Data<long>(), 0, dotHandle, nd.size);
537+
Marshal.Copy(nd1.ToArray<long>(), 0, dotHandle, nd.size);
538538
break;
539539
case "Single":
540-
Marshal.Copy(nd1.Data<float>(), 0, dotHandle, nd.size);
540+
Marshal.Copy(nd1.ToArray<float>(), 0, dotHandle, nd.size);
541541
break;
542542
case "Double":
543-
Marshal.Copy(nd1.Data<double>(), 0, dotHandle, nd.size);
543+
Marshal.Copy(nd1.ToArray<double>(), 0, dotHandle, nd.size);
544544
break;
545545
case "Byte":
546-
Marshal.Copy(nd1.Data<byte>(), 0, dotHandle, nd.size);
546+
Marshal.Copy(nd1.ToArray<byte>(), 0, dotHandle, nd.size);
547547
break;
548548
case "String":
549-
return new Tensor(UTF8Encoding.UTF8.GetBytes(nd.Data<string>(0)), TF_DataType.TF_STRING);
549+
throw new NotImplementedException($"Marshal.Copy failed for {nd.dtype.Name}.");
550+
//return new Tensor(UTF8Encoding.UTF8.GetBytes(nd.ToArray<string>(0)), TF_DataType.TF_STRING);
550551
default:
551552
throw new NotImplementedException($"Marshal.Copy failed for {nd.dtype.Name}.");
552553
}

src/TensorFlowNET.Core/Tensors/Tensor.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ public unsafe string[] StringData()
196196
// [offset1, offset2,...,offsetn, s1size, s1bytes, s2size, s2bytes,...,snsize,snbytes]
197197
//
198198
long size = 1;
199-
foreach (var s in TensorShape.Dimensions)
199+
foreach (var s in TensorShape.dims)
200200
size *= s;
201201

202202
var buffer = new byte[size][];

test/TensorFlowNET.Examples/ImageProcessing/DigitRecognitionNN.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ public void PrepareData()
127127
public void Train(Session sess)
128128
{
129129
// Number of training iterations in each epoch
130-
var num_tr_iter = mnist.Train.Labels.len / batch_size;
130+
var num_tr_iter = mnist.Train.Labels.shape[0] / batch_size;
131131

132132
var init = tf.global_variables_initializer();
133133
sess.run(init);

0 commit comments

Comments
 (0)