fix bug

PaddlePaddle · Aug 20, 2022 · 8354111 · 8354111
1 parent 5a4fdc6
commit 8354111
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cinn/frontend/net_builder.cc b/cinn/frontend/net_builder.cc
@@ -88,7 +88,7 @@ Variable NetBuilder::Gather(const Variable& x, const Variable& index, const int&
 
 Variable NetBuilder::GatherNd(const Variable& x, const Variable& index, const std::vector<int>& axes) {
   Instruction instr("gather", {x, index});
-  instr.SetAttr("axis", axis);
+  instr.SetAttr("axis", axes);
   InferShape(instr);
   AppendInstruction(instr);
   return instr.GetOutput(0);