Commit 2631f50f by xiaotong

合并分支 'xiaotong-working' 到 'master'

bug fix for Select

查看合并请求 NiuTrans/NiuTrans.Tensor!3
parents 69a31b42 699f5116
......@@ -52,6 +52,5 @@ int main( int argc, const char ** argv )
fprintf(stderr, "Run this program with \"-test\" for unit test!\n");
fprintf(stderr, "Or run this program with \"-fnnlm\" for sample FNNLM!\n");
}
// haha
return 0;
}
......@@ -55,7 +55,7 @@ void SelectRange(XTensor * a, int dim, int low, int high, XTensor * c)
}
int stride = 1;
for(int i = 0; i < a->order; i++)
for(int i = 0; i < dim; i++)
stride *= a->dimSizeRDI[i];
int blockSize = stride * (high - low) * a->unitSize;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论