mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!46328 fix bug in maxunpool1d. 

Merge pull request !46328 from ZhidanLiu/master
This commit is contained in:
i-robot 2022-12-02 08:03:40 +00:00 committed by Gitee
parent 36a74bc62e 36738babc0
commit 9eef6c4e0e
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/api/api_python/ops/mindspore.ops.func_tensor_split.rst
View File

@ -19,7 +19,7 @@ mindspore.ops.tensor_split
- **TypeError** - `x` 不是Tensor。
- **TypeError** - `axis` 不是int类型。
- **TypeError** - `axis` 不是int类型。
- **ValueError** - 参数 `axis` 超出 :math:`-x.dim, x.dim)` 范围。
- **ValueError** - 参数 `axis` 超出 :math:`[-x.dim, x.dim)` 范围。
- **TypeError** - `indices_or_sections` 中的每个元素不是int类型
- **TypeError** - `indices_or_sections` 不是int，tuple(int)或list(int)。

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

@ -743,7 +743,7 @@ def max_unpool1d(x, indices, kernel_size, stride=None, padding=0, output_size=No
if len(output_size) == 2:
output_size = (1,) + output_size + (1,)
if len(output_size) == 3:
output_size = (1,) + output_size
output_size = output_size + (1,)
shape = P.Shape()
x_shape = shape(x)