|
|
|
|
|
|
|
|
|
|
public void Generate(TensorProxy tensorProxy, int batchSize, Dictionary<Agent, AgentInfo> agentInfo) |
|
|
|
{ |
|
|
|
tensorProxy.data?.Dispose(); |
|
|
|
tensorProxy.data = _allocator.Alloc(new TensorShape(1,1)); |
|
|
|
tensorProxy.data[0] = batchSize; |
|
|
|
} |
|
|
|
|
|
|
public void Generate(TensorProxy tensorProxy, int batchSize, Dictionary<Agent, AgentInfo> agentInfo) |
|
|
|
{ |
|
|
|
tensorProxy.shape = new long[0]; |
|
|
|
tensorProxy.data?.Dispose(); |
|
|
|
|
|
|
|
tensorProxy.data[0] = 1; |
|
|
|
} |
|
|
|
} |
|
|
|