【PaddlePaddle Hackathon 3】16 新增 API paddle.take

docs/api/api_label

@@ -246,6 +246,7 @@ paddle.subtract	.. _api_paddle_subtract:
 paddle.Tensor.subtract_	.. _api_paddle_Tensor_subtract_:
 paddle.sum	.. _api_paddle_sum:
 paddle.t	.. _api_paddle_t:
+paddle.take	.. _api_paddle_take:
 paddle.tanh	.. _api_paddle_tanh:
 paddle.tanh_	.. _api_paddle_tanh_:
 paddle.tile	.. _api_paddle_tile:

docs/api/paddle/Overview_cn.rst

@@ -136,6 +136,7 @@ tensor 数学操作
     " :ref:`paddle.acosh <cn_api_fluid_layers_acosh>` ", "反双曲余弦函数"
     " :ref:`paddle.asinh <cn_api_fluid_layers_asinh>` ", "反双曲正弦函数"
     " :ref:`paddle.atanh <cn_api_fluid_layers_atanh>` ", "反双曲正切函数"
+    " :ref:`paddle.take <cn_api_tensor_take>` ", "输出给定索引处的输入元素，结果与 index 的形状相同" 
 
 
 .. _tensor_logic:

docs/api/paddle/Tensor/Overview_en.rst

@@ -246,6 +246,7 @@ Methods
     subtract_
     sum
     t
+    take
     tanh
     tanh_
     tile

docs/api/paddle/Tensor_cn.rst

@@ -2412,3 +2412,12 @@ erfinv(x, name=None)
 对输入 x 进行逆误差函数计算
 
 请参考 :ref:`cn_api_paddle_tensor_erfinv`
+
+take(input, index, name=None)
+:::::::::
+
+返回：一个新的 Tensor，其中包含给定索引处的输入元素。结果与 `index` 的形状相同
+
+返回类型：Tensor
+
+请参考 :ref:`cn_api_tensor_take`

docs/api/paddle/take_cn.rst

@@ -0,0 +1,28 @@
+.. _cn_api_paddle_tensor_take:
+
+take
+-------------------------------
+
+.. py:function:: paddle.take(input, index, name=None)
+
+返回一个新的 Tensor，其中包含给定索引处的输入元素。
+将输入 Tensor 视为一维 Tensor，返回指定索引上的元素集合，返回结果与 `index` 的形状相同。
+
+参数
+:::::::::
+
+- **input**  (Tensor) - 输入的 Tensor，支持 int32、int64、float32、float64 数据类型。
+- **index**  (Tensor) - 索引矩阵，支持 int32、int64 数据类型。
+- **name**  (str，可选) - 具体用法请参见 :ref:`api_guide_Name`，一般无需设置，默认值为 None。
+
+返回
+:::::::::
+
+- Tensor - 返回一个新的 Tensor，其中包含给定索引处的输入元素。结果与 index 的形状相同。
+
+代码示例
+:::::::::
+
+
+COPY-FROM: paddle.take
+

docs/api/paddle/vision/models/alexnet_cn.rst

@@ -1,18 +1,17 @@
-.. _cn_api_paddle_vision_models_alexnet:
+.. _cn_api_paddle_vision_models_AlexNet:
 
-alexnet
+AlexNet
 -------------------------------
 
-.. py:function:: paddle.vision.models.alexnet(pretrained=False, **kwargs)
+.. py:function:: paddle.vision.models.AlexNet(num_classes=1000)
 
 
 AlexNet 模型，来自论文 `"ImageNet Classification with Deep Convolutional Neural Networks" <https://papers.nips.cc/paper/2012/file/c399862d3b9d6b76c8436e924a68c45b-Paper.pdf>`_ 。
 
 参数
 :::::::::
 
-  - **pretrained** (bool，可选) - 是否加载预训练权重。如果为 True，则返回在 ImageNet 上预训练的模型。默认值为 False。
-  - **\*\*kwargs** (可选) - 附加的关键字参数，具体可选参数请参见 :ref:`AlexNet <cn_api_paddle_vision_models_AlexNet>`。
+  - **num_classes** (int，可选) - 最后一个全连接层输出的维度。如果该值小于等于 0，则不定义最后一个全连接层。默认值为 1000。
 
 返回
 :::::::::
@@ -22,4 +21,4 @@ AlexNet 模型，来自论文 `"ImageNet Classification with Deep Convolutional
 代码示例
 :::::::::
 
-COPY-FROM: paddle.vision.models.alexnet
+COPY-FROM: paddle.vision.models.AlexNet

docs/api/paddle/vision/models/googlenet_cn.rst

@@ -1,18 +1,18 @@
-.. _cn_api_paddle_vision_models_googlenet:
+.. _cn_api_paddle_vision_models_GoogLeNet:
 
-googlenet
+GoogLeNet
 -------------------------------
 
-.. py:function:: paddle.vision.models.googlenet(pretrained=False, **kwargs)
+.. py:function:: paddle.vision.models.googlenet(num_classes=1000, with_pool=True)
 
 
 GoogLeNet（Inception v1）模型，来自论文 `"Going Deeper with Convolutions" <https://arxiv.org/pdf/1409.4842.pdf>`_ 。
 
 参数
 :::::::::
 
-  - **pretrained** (bool，可选) - 是否加载预训练权重。如果为 True，则返回在 ImageNet 上预训练的模型。默认值为 False。
-  - **\*\*kwargs** (可选) - 附加的关键字参数，具体可选参数请参见 :ref:`GoogLeNet <cn_api_paddle_vision_models_GoogLeNet>`。
+  - **num_classes** (int，可选) - 最后一个全连接层输出的维度。如果该值小于等于 0，则不定义最后一个全连接层。默认值为 1000。
+  - **with_pool** (bool，可选) - 是否定义最后一个全连接层之前的池化层。默认值为 True。
 
 返回
 :::::::::
@@ -22,4 +22,4 @@ GoogLeNet（Inception v1）模型，来自论文 `"Going Deeper with Convolution
 代码示例
 :::::::::
 
-COPY-FROM: paddle.vision.models.googlenet
+COPY-FROM: paddle.vision.models.GoogLeNet