Class ArrayTensorData
Internal Tensor data backed by managed array
Inherited Members
Namespace: Unity.Barracuda
Assembly: Unity.Barracuda.dll
Syntax
public class ArrayTensorData : UniqueResourceId, ITensorData, IDisposable, ITensorDataStatistics, IUniqueResource
Constructors
ArrayTensorData(int, DataType)
Create ArrayTensorData and allocate storage for count elements
Declaration
public ArrayTensorData(int count, DataType dataType = DataType.Float)
Parameters
| Type | Name | Description |
|---|---|---|
| int | count | number of elements to pre-allocate |
| DataType | dataType |
ArrayTensorData(TensorShape, DataType)
Create ArrayTensorData and allocate storage for Tensor described by shape
Declaration
public ArrayTensorData(TensorShape shape, DataType dataType = DataType.Float)
Parameters
| Type | Name | Description |
|---|---|---|
| TensorShape | shape | shape |
| DataType | dataType |
Properties
array
Data storage array
Declaration
public BarracudaArray array { get; }
Property Value
| Type | Description |
|---|---|
| BarracudaArray |
dataType
Returns the type of the elements this tensorData can contain.
Declaration
public virtual DataType dataType { get; }
Property Value
| Type | Description |
|---|---|
| DataType |
inUse
Declaration
public virtual bool inUse { get; }
Property Value
| Type | Description |
|---|---|
| bool |
isGPUMem
Declaration
public virtual bool isGPUMem { get; }
Property Value
| Type | Description |
|---|---|
| bool |
maxCapacity
Returns the maximum number of element this tensorData can contain.
Declaration
public virtual int maxCapacity { get; }
Property Value
| Type | Description |
|---|---|
| int |
Methods
Dispose()
Dispose storage
Declaration
public virtual void Dispose()
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
public virtual 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 |
~ArrayTensorData()
Finalizer
Declaration
protected ~ArrayTensorData()
Reserve(int)
Reserve uninitialized memory.
Declaration
public virtual 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
public virtual bool ScheduleAsyncDownload(int count)
Parameters
| Type | Name | Description |
|---|---|---|
| int | count | count of elements to download |
Returns
| Type | Description |
|---|---|
| bool |
|
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
public virtual BarracudaArray SharedAccess(out int offset)
Parameters
| Type | Name | Description |
|---|---|---|
| int | offset | This function outputs |
Returns
| Type | Description |
|---|---|
| BarracudaArray | array filled with the values of multiple tensors that share the same tensorData on device |
ToString()
Storage summary as string
Declaration
public override string ToString()
Returns
| Type | Description |
|---|---|
| string | storage summary as string |
Overrides
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
public virtual void Upload(float[] data, TensorShape shape, int managedBufferStartIndex = 0)
Parameters
| Type | Name | Description |
|---|---|---|
| float[] | data | data as |
| TensorShape | shape | Tensor shape |
| int | managedBufferStartIndex | managed buffer start index |