xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix gather

This commit is contained in:
ckey_Dou 2024-03-21 20:57:35 +08:00 committed by hedongdong
parent 2dff16022b
commit 772542e148
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/ops/function/array_func.py
View File

@ -2800,7 +2800,7 @@ def gather_elements(input, dim, index):
int32, int64. The value range of each index element is `[-input.shape(dim), input.shape(dim))`. int32, int64. The value range of each index element is `[-input.shape(dim), input.shape(dim))`.
Returns: Returns:
Tensor, has the same shape as `index` tensor and has the same data type with `input`. Tensor, has the same shape as `index` and has the same data type with `input`.
Raises: Raises:
TypeError: If dtype of `dim` or `index` is neither int32 nor int64. TypeError: If dtype of `dim` or `index` is neither int32 nor int64.