diff --git a/paddle/phi/core/string_tensor.cc b/paddle/phi/core/string_tensor.cc index 123ca74a631f4..2acd0e964d576 100644 --- a/paddle/phi/core/string_tensor.cc +++ b/paddle/phi/core/string_tensor.cc @@ -108,6 +108,19 @@ StringTensor& StringTensor::Resize(const DDim& dims) { meta_.dims = dims; return *this; } + +StringTensor& StringTensor::Resize(const std::initializer_list dims) { + return Resize(make_ddim(dims)); +} + +StringTensor& StringTensor::Resize(const std::vector& dims) { + return Resize(make_ddim(dims)); +} + +StringTensor& StringTensor::Resize(const std::vector& dims) { + return Resize(make_ddim(dims)); +} + // TODO(zhoushunjie): need to remove it for general space void StringTensor::init_holder() { void* ptr = holder_->ptr(); diff --git a/paddle/phi/core/string_tensor.h b/paddle/phi/core/string_tensor.h index 35b2f7b64d20f..a3245de53597b 100644 --- a/paddle/phi/core/string_tensor.h +++ b/paddle/phi/core/string_tensor.h @@ -112,6 +112,12 @@ class PADDLE_API StringTensor StringTensor& Resize(const DDim& dims); + StringTensor& Resize(const std::initializer_list dims); + + StringTensor& Resize(const std::vector& dims); + + StringTensor& Resize(const std::vector& dims); + /// \brief Returns the actual storage size occupied by tensor, may be larger /// than its shape dims. /// \return The actual storage size occupied by tensor. diff --git a/paddle/phi/kernels/strings/strings_empty_kernel.cc b/paddle/phi/kernels/strings/strings_empty_kernel.cc index 36c9e33cff1fb..2eeb49207f1f7 100644 --- a/paddle/phi/kernels/strings/strings_empty_kernel.cc +++ b/paddle/phi/kernels/strings/strings_empty_kernel.cc @@ -23,7 +23,7 @@ template void EmptyKernel(const Context& dev_ctx, const IntArray& shape, StringTensor* out) { - out->Resize(make_ddim(shape.GetData())); + out->Resize(shape.GetData()); dev_ctx.template Alloc(out); }