|
|
|
|
|
|
|
|
|
|
TensorShape m_TensorShape; |
|
|
|
|
|
|
|
internal WriteAdapter() { } |
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
/// Set the adapter to write to an IList at the given channelOffset.
|
|
|
|
/// </summary>
|
|
|
|
|
|
|
public void SetTarget(IList<float> data, int[] shape, int offset) |
|
|
|
internal void SetTarget(IList<float> data, int[] shape, int offset) |
|
|
|
{ |
|
|
|
m_Data = data; |
|
|
|
m_Offset = offset; |
|
|
|
|
|
|
/// <param name="tensorProxy">Tensor proxy that will be writtent to.</param>
|
|
|
|
/// <param name="batchIndex">Batch index in the tensor proxy (i.e. the index of the Agent)</param>
|
|
|
|
/// <param name="channelOffset">Offset from the start of the channel to write to.</param>
|
|
|
|
public void SetTarget(TensorProxy tensorProxy, int batchIndex, int channelOffset) |
|
|
|
internal void SetTarget(TensorProxy tensorProxy, int batchIndex, int channelOffset) |
|
|
|
{ |
|
|
|
m_Proxy = tensorProxy; |
|
|
|
m_Batch = batchIndex; |
|
|
|