|
|
|
@ -74,11 +74,10 @@ Tensor CombineBatchAndN(const framework::ExecutionContext& context,
|
|
|
|
|
Tensor output;
|
|
|
|
|
auto in_dims = input.dims();
|
|
|
|
|
if (in_dims.size() == 3) {
|
|
|
|
|
output.Resize(in_dims);
|
|
|
|
|
output.Resize({in_dims[1], in_dims[0], in_dims[2]});
|
|
|
|
|
output.mutable_data<T>(context.GetPlace());
|
|
|
|
|
EigenTranspose<Place, T, 3>(context, input, output, {1, 0, 2});
|
|
|
|
|
std::vector<int64_t> out_dims = {in_dims[1], in_dims[0] * in_dims[2]};
|
|
|
|
|
output.Resize(make_ddim(out_dims));
|
|
|
|
|
output.Resize({in_dims[1], in_dims[0] * in_dims[2]});
|
|
|
|
|
} else {
|
|
|
|
|
output.ShareDataWith(input);
|
|
|
|
|
}
|
|
|
|
|