From 767d0ba26732a002cce27563f989a4514f1c7029 Mon Sep 17 00:00:00 2001
From: lilong12 <lilong12@baidu.com>
Date: Tue, 24 Nov 2020 14:39:59 +0800
Subject: [PATCH] update, test=develop (#28700)

---
 paddle/fluid/operators/expand_as_v2_op.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/expand_as_v2_op.cc b/paddle/fluid/operators/expand_as_v2_op.cc
index 495b640bb4..e8008056c4 100644
--- a/paddle/fluid/operators/expand_as_v2_op.cc
+++ b/paddle/fluid/operators/expand_as_v2_op.cc
@@ -45,7 +45,7 @@ class ExpandAsV2Op : public framework::OperatorWithKernel {
             "The rank of Input(target_tensor) must not be less than or equal "
             "to %d. But received: input rank %u, input shape [%s].",
             MAX_RANK_SUPPORTED, x_dims.size(), x_dims));
-    std::vector<int64_t> out_shape(target_tensor_dims.size());
+    std::vector<int> out_shape = framework::vectorize<int>(target_tensor_dims);
     ctx->SetOutputDim("Out", framework::make_ddim(out_shape));
   }
 };
-- 
GitLab