TensorFlow.NET/src/TensorFlowNET.Core/Tensors/Tensor.Value.cs at master · ehtick/TensorFlow.NET · GitHub

History

75 lines (65 loc) · 2.17 KB

Raw

using Tensorflow.NumPy;

using System;

using System.Text;

using static Tensorflow.Binding;

namespace Tensorflow

{

public partial class Tensor

{

/// <summary>

///

/// </summary>

/// <typeparam name="T"></typeparam>

/// <returns></returns>

public virtual unsafe T[] ToArray<T>() where T : unmanaged

{

//Are the types matching?

if (typeof(T).as_tf_dtype() != dtype)

throw new ArrayTypeMismatchException($"Required dtype {dtype} mismatch with {typeof(T).as_tf_dtype()}.");

if (ndim == 0 && size == 1) //is it a scalar?

{

unsafe

{

return new T[] { *(T*)buffer };

}

//types match, no need to perform cast

var ret = new T[size];

var len = (long)(size * dtypesize);

var src = (T*)buffer;

fixed (T* dst = ret)

System.Buffer.MemoryCopy(src, dst, len, len);

return ret;

}

/// <summary>

/// Copy of the contents of this Tensor into a NumPy array or scalar.

/// </summary>

/// <returns>

/// A NumPy array of the same shape and dtype or a NumPy scalar, if this

/// Tensor has rank 0.

/// </returns>

public NDArray numpy()

=> GetNDArray(dtype);

protected NDArray GetNDArray(TF_DataType dtype)

{

if (dtype == TF_DataType.TF_STRING)

{

var str= StringData();

return new NDArray(str, shape);

}

return new NDArray(this, clone: true);

}

/// <summary>

/// Copies the memory of current buffer onto newly allocated array.

/// </summary>

/// <returns></returns>

public unsafe byte[] BufferToArray()

{

// ReSharper disable once LocalVariableHidesMember

var data = new byte[bytesize];

fixed (byte* dst = data)

System.Buffer.MemoryCopy(buffer.ToPointer(), dst, bytesize, bytesize);

return data;

}

TensorFlow.NET/src/TensorFlowNET.Core/Tensors/Tensor.Value.cs at master · ehtick/TensorFlow.NET · GitHub

Expand file tree

Breadcrumbs

Tensor.Value.cs

Latest commit

History

Breadcrumbs

Tensor.Value.cs

File metadata and controls