Interface ITensorData

Interface for device dependent representation of Tensor data.

Inherited Members

IDisposable.Dispose()

ITensorDataStatistics.maxCapacity

ITensorDataStatistics.dataType

Namespace: Unity.Barracuda

Assembly: solution.dll

Syntax

public interface ITensorData : IDisposable, ITensorDataStatistics, IUniqueResource

Methods

Download(TensorShape)

Returns an array filled with the values of a tensor. Depending on the implementation and underlying device this array might be a copy or direct reference to the tensor values. This is a blocking call, unless data from device was requested via ScheduleAsyncDownload beforehand and has already arrived.

Declaration

float[] Download(TensorShape shape)

Parameters

Type	Name	Description
TensorShape	shape	the TensorShape (and thus length) of the data to copy

Returns

Type	Description
float[]	Tensor data as float arrary

Reserve(int)

Reserve uninitialized memory.

Declaration

void Reserve(int count)

Parameters

Type	Name	Description
int	count	element count to reserve

ScheduleAsyncDownload(int)

Schedule an asynchronous download from device memory. count is the number of element to readback.

Declaration

bool ScheduleAsyncDownload(int count)

Parameters

Type	Name	Description
int	count	count of elements to download

Returns

Type	Description
bool	false until data from device arrives to CPU and is ready for access

SharedAccess(out int)

Returns an array filled with the values of multiple tensors that share the same tensorData on device. Depending on the implementation and underlying device this array might be a copy or direct reference to tensor values, no conversion from on device memory layout will occur. This is a blocking call, unless data from device was requested via ScheduleAsyncDownload beforehand and has already arrived.

Declaration

BarracudaArray SharedAccess(out int offset)

Parameters

Type	Name	Description
int	offset	This function outputs offset from the beginning of the array to location of values for specific tensor. offset parameters is specified in float elements

Returns

Type	Description
BarracudaArray	array filled with the values of multiple tensors that share the same tensorData on device

Upload(float[], TensorShape, int)

Initialize with data. shape is the TensorShape (and thus length) of the data to copy. managedBufferStartIndex is the offset where to start the copy in the data

Declaration

void Upload(float[] data, TensorShape shape, int managedBufferStartIndex = 0)

Parameters

Type	Name	Description
float[]	data	data as float array
TensorShape	shape	Tensor shape
int	managedBufferStartIndex	managed buffer start index

Extension Methods

DeprecatedTensorDataExtensions.GetMaxCount(ITensorData)