File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change 4040
4141参数
4242::::::::::::
43- - **x ** (Tensor) - 输入 Tensor,秩 ``rank >= 1 `` ,支持的数据类型包括 int32、int64、float32、float64、complex64、complex128 和 uint8 (CPU)、float16(GPU) 。
44- - **index ** (Tensor) - 索引 Tensor,秩 ``rank = 0 `` 或者 ``rank = 1 `` ,数据类型为 int32 或 int64。
45- - **axis ** (Tensor) - 指定 index 获取输入的维度, ``axis `` 的类型可以是 int 或者 Tensor,当 ``axis `` 为 Tensor 的时候其数据类型为 int32 或者 int64。默认值为 None,当 ``axis `` 为 None 的时候其值为 0。
43+ - **x ** (Tensor) - 输入 Tensor,秩 ``rank >= 1 ``,支持的数据类型包括 int32、int64、float32、float64、complex64、complex128 和 uint8 (CPU)、float16(GPU) 。
44+ - **index ** (Tensor) - 索引 Tensor,秩 ``rank = 0 `` 或者 ``rank = 1 ``,数据类型为 int32 或 int64。
45+ - **axis ** (Tensor) - 指定 index 获取输入的维度,``axis `` 的类型可以是 int 或者 Tensor,当 ``axis `` 为 Tensor 的时候其数据类型为 int32 或者 int64。默认值为 None,当 ``axis `` 为 None 的时候其值为 0。
4646 - **name ** (str,可选) - 具体用法请参见 :ref: `api_guide_Name `,一般无需设置,默认值为 None。
4747
4848返回
@@ -68,13 +68,13 @@ PyTorch 兼容的 ``gather`` 操作:根据索引 index 获取输入 ``input``
6868参数
6969::::::::::::
7070 - **input ** (Tensor) - 输入 Tensor,支持的数据类型包括 int32、int64、float32、float64、int16、uint8、float16(GPU)以及 bfloat16(GPU) 。
71- - **dim ** (int) - 指定 index 获取输入的维度, ``dim `` 的类型可以是 int 或者。
72- - **index ** (Tensor) - 索引 Tensor, ``index `` 张量的各维度需要小于等于 ``input `` 张量的各维度(除 ``dim `` 维度外),且值需要在 ``input.shape[dim] `` 范围内。数据类型为 int32 或 int64。
71+ - **dim ** (int) - 指定 index 获取输入的维度,``dim `` 的类型可以是 int 或者。
72+ - **index ** (Tensor) - 索引 Tensor,``index `` 张量的各维度需要小于等于 ``input `` 张量的各维度(除 ``dim `` 维度外),且值需要在 ``input.shape[dim] `` 范围内。数据类型为 int32 或 int64。
7373 - **out ** (Tensor,可选) - 用于引用式传入输出值,注意:动态图下 out 可以是任意 Tensor,默认值为 None。
7474
7575.. caution ::
7676
77- 本接口没有实现 PyTorch 的 ``sparse_grad `` 参数!梯度默认是稠密的,等效于 ``sparse_grad=False `` 。
77+ 本接口没有实现 PyTorch 的 ``sparse_grad `` 参数!梯度默认是稠密的,等效于 ``sparse_grad=False ``。
7878
7979
8080返回
You can’t perform that action at this time.
0 commit comments