From 6bfbb6090214af45684d6533045ca17ba57df4ac Mon Sep 17 00:00:00 2001 From: LaJiOJ <1184264181@qq.com> Date: Wed, 18 Aug 2021 11:27:34 +0800 Subject: [PATCH 1/2] Add paddle::Tensor::slice API doc --- docs/guides/07_new_op/new_custom_op_cn.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docs/guides/07_new_op/new_custom_op_cn.md b/docs/guides/07_new_op/new_custom_op_cn.md index 320ad4c3642..402fceb030a 100644 --- a/docs/guides/07_new_op/new_custom_op_cn.md +++ b/docs/guides/07_new_op/new_custom_op_cn.md @@ -97,6 +97,8 @@ std::vector OpFucntion(const paddle::Tensor& x, ..., int attr, . - 模板类方法，获取数据内存的起始地址（只读访问） - `template T* mutable_data(const PlaceType& place)`： - 模板类方法，输入参数 `place` ，根据 `Tensor.shape` 在指定设备上申请内存，并返回内存的起始地址 + - `Tensor slice(const int64_t begin_idx, const int64_t end_idx) const`： + - 输入参数起始行 `begin_idx` 和终止行 `end_idx`，返回当前 `Tensor` 从起始行（含）到终止行（不含）的视图 - 工具类API： - `template Tensor copy_to(const PlaceType& place) const`： - 模板类方法，输入参数 `place`，将当前 `Tensor` 拷贝到指定设备上并返回 From 966ec803245357331a4c245398aed4a0cf641e5b Mon Sep 17 00:00:00 2001 From: LaJiOJ Date: Wed, 18 Aug 2021 16:24:41 +0800 Subject: [PATCH 2/2] Add note for slice api --- docs/guides/07_new_op/new_custom_op_cn.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/guides/07_new_op/new_custom_op_cn.md b/docs/guides/07_new_op/new_custom_op_cn.md index 402fceb030a..61e3310f97a 100644 --- a/docs/guides/07_new_op/new_custom_op_cn.md +++ b/docs/guides/07_new_op/new_custom_op_cn.md @@ -98,7 +98,8 @@ std::vector OpFucntion(const paddle::Tensor& x, ..., int attr, . - `template T* mutable_data(const PlaceType& place)`： - 模板类方法，输入参数 `place` ，根据 `Tensor.shape` 在指定设备上申请内存，并返回内存的起始地址 - `Tensor slice(const int64_t begin_idx, const int64_t end_idx) const`： - - 输入参数起始行 `begin_idx` 和终止行 `end_idx`，返回当前 `Tensor` 从起始行（含）到终止行（不含）的视图 + - 输入参数起始行 `begin_idx` 和终止行 `end_idx`，返回当前 `Tensor` 从起始行（含）到终止行（不含）的一个视图 + > 注：本API仅支持对当前 `Tensor` 的第一个维度（即 axis = 0）进行切分 - 工具类API： - `template Tensor copy_to(const PlaceType& place) const`： - 模板类方法，输入参数 `place`，将当前 `Tensor` 拷贝到指定设备上并返回