From 772542e148c648e0120b13b22417459dec63bbc9 Mon Sep 17 00:00:00 2001 From: ckey_Dou Date: Thu, 21 Mar 2024 20:57:35 +0800 Subject: [PATCH] fix gather --- mindspore/python/mindspore/ops/function/array_func.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/python/mindspore/ops/function/array_func.py b/mindspore/python/mindspore/ops/function/array_func.py index fc7e7abec3c..78cd1070c8d 100644 --- a/mindspore/python/mindspore/ops/function/array_func.py +++ b/mindspore/python/mindspore/ops/function/array_func.py @@ -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))`. 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: TypeError: If dtype of `dim` or `index` is neither int32 nor int64.