Class ArrayTensorData
Internal Tensor
data backed by managed array
Inherited Members
Namespace: Unity.Barracuda
Assembly: solution.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
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 |
---|---|---|
Tensor |
shape | shape |
Data |
dataType |
Properties
array
Data storage array
Declaration
public BarracudaArray array { get; }
Property Value
Type | Description |
---|---|
Barracuda |
dataType
Returns the type of the elements this tensorData can contain.
Declaration
public virtual DataType dataType { get; }
Property Value
Type | Description |
---|---|
Data |
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 |
---|---|---|
Tensor |
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 |
---|---|
Barracuda |
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 |
Tensor |
shape | Tensor shape |
int | managedBufferStartIndex | managed buffer start index |