/// <summary>
/// Allocate a new tensor. The caller must set the Tensor values by writing them to the DataPointer
/// with length ByteSize.
/// </summary>
/// <param name="dataType">The type of data</param>
/// <param name="dims">The size for each of the dimension of the tensor</param>
public Tensor(DataType dataType, int[] dims)
{
int size = dims.Length > 0 ? 1 : 0;
for (int i = 0; i < dims.Length; i++)
{
size *= dims[i];
}
size *= TfInvoke.DataTypeSize(dataType);
GCHandle dimHandle = GCHandle.Alloc(dims, GCHandleType.Pinned);
_ptr = TfInvoke.tfeAllocateTensor(dataType, dimHandle.AddrOfPinnedObject(), dims.Length, size);
dimHandle.Free();
}
/// <summary>
/// Create a single element tensor
/// </summary>
/// <param name="dataType">The Type of the Tensor</param>
/// <param name="sizeInBytes">The size in bytes</param>
private Tensor(DataType dataType, int sizeInBytes)
{
_ptr = TfInvoke.tfeAllocateTensor(dataType, IntPtr.Zero, 0, sizeInBytes);
}
