提交 12d17941 编写于 作者: F fengjiayi

fix a bug

上级 41ced8e2
...@@ -24,8 +24,8 @@ struct ReverseFunctor { ...@@ -24,8 +24,8 @@ struct ReverseFunctor {
void operator()(const DeviceContext& context, const framework::LoDTensor& in, void operator()(const DeviceContext& context, const framework::LoDTensor& in,
framework::LoDTensor* out, const std::vector<int>& axis) { framework::LoDTensor* out, const std::vector<int>& axis) {
Eigen::array<bool, Rank> reverse_axis; Eigen::array<bool, Rank> reverse_axis;
for (auto& i : reverse_axis) { for (int i = 0; i < Rank; ++i) {
i = false; reverse_axis[i] = false;
} }
for (int a : axis) { for (int a : axis) {
reverse_axis[a] = true; reverse_axis[a] = true;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册