diff --git a/paddle/framework/ddim.h b/paddle/framework/ddim.h index 675f8680f699cf52d6653ec51f6909f4be8e07fc..df26d73d481e85bcfcbd41dd1cde446b311cb595 100644 --- a/paddle/framework/ddim.h +++ b/paddle/framework/ddim.h @@ -81,6 +81,13 @@ std::vector vectorize(const DDim& ddim); ssize_t product(const DDim& ddim); +/** + * \brief Slice a ddim + * + * Slice dim with [begin, end). + * e.g. DDim d = make_ddim({1,2,3,4,5}); + * slice_ddim(d, 1, 3); ====> {2,3} + */ DDim slice_ddim(const DDim& dim, int begin, int end); /**