Browse Source

Tensor.BufferToArray(): Fixed to use long instead of int.

tags/v0.12
Eli Belash 6 years ago
parent
commit
f375e1c0bb
1 changed files with 10 additions and 3 deletions
  1. +10
    -3
      src/TensorFlowNET.Core/Tensors/Tensor.cs

+ 10
- 3
src/TensorFlowNET.Core/Tensors/Tensor.cs View File

@@ -372,9 +372,16 @@ namespace Tensorflow
/// <returns></returns>
public byte[] BufferToArray()
{
var data = new byte[bytesize];
Marshal.Copy(buffer, data, 0, (int) bytesize);
return data;
unsafe
{
// ReSharper disable once LocalVariableHidesMember
var bytesize = (long) this.bytesize;
var data = new byte[bytesize];
fixed (byte* dst = data)
System.Buffer.MemoryCopy(buffer.ToPointer(), dst, bytesize, bytesize);

return data;
}
}

/// Used internally in ToArray&lt;T&gt;


Loading…
Cancel
Save