提交 d94e1f51 编写于 作者: C Cao Ying 提交者: GitHub

Merge pull request #1754 from lcy-seso/fix_bug_of_AgentLayer

fix a bug of AgentLayer.
...@@ -42,7 +42,8 @@ void AgentLayer::forward(PassType passType) { ...@@ -42,7 +42,8 @@ void AgentLayer::forward(PassType passType) {
// get Arguments from real layers // get Arguments from real layers
if (numSamples_ > 0 && numSamples_ < realHeight) { if (numSamples_ > 0 && numSamples_ < realHeight) {
if (realOutput.ids) { if (realOutput.ids) {
output_.ids->subVecFrom(*realOutput.ids, 0, numSamples_); output_.ids =
IVector::create(realOutput.ids->getData(), numSamples_, useGpu_);
} else { } else {
output_.subArgFrom( output_.subArgFrom(
realOutput, /* offset */ 0, numSamples_, getSize(), useGpu_); realOutput, /* offset */ 0, numSamples_, getSize(), useGpu_);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册