Merge pull request from lcy-seso/fix_bug_of_AgentLayer

fix a bug of AgentLayer.
release/0.10.0
Cao Ying 8 years ago committed by GitHub
commit d94e1f516c

@ -42,7 +42,8 @@ void AgentLayer::forward(PassType passType) {
// get Arguments from real layers
if (numSamples_ > 0 && numSamples_ < realHeight) {
if (realOutput.ids) {
output_.ids->subVecFrom(*realOutput.ids, 0, numSamples_);
output_.ids =
IVector::create(realOutput.ids->getData(), numSamples_, useGpu_);
} else {
output_.subArgFrom(
realOutput, /* offset */ 0, numSamples_, getSize(), useGpu_);

Loading…
Cancel
Save