未验证 提交 767d0ba2 编写于 作者: L lilong12 提交者: GitHub

update, test=develop (#28700)

上级 85292e0b
......@@ -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));
}
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册