fix gaussion op bug

8 years ago · b054392e2a
parent 08c987d7c0
commit b054392e2a
1 changed files with 1 additions and 1 deletions
--- a/paddle/operators/gaussian_random_op.cc
+++ b/paddle/operators/gaussian_random_op.cc
@ -23,7 +23,7 @@ class GaussianRandomOp : public framework::OperatorWithKernel {

 protected:
  void InferShape(const framework::InferShapeContext& context) const override {
-    auto* tensor = context.Output<framework::Tensor>(0);
+    auto* tensor = context.Output<framework::Tensor>("Out");
    auto dims = GetAttr<std::vector<int>>("dims");
    PADDLE_ENFORCE(dims.size() > 0UL,
                   "dims can be one int or array. dims must be set.");