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

fix bug in maxunpool1d.

This commit is contained in:
ZhidanLiu 2022-12-01 19:28:57 +08:00
parent 40e57737ee
commit 36738babc0
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)