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

!51693 fix unfold doc example 

Merge pull request !51693 from baihuawei/code_docs_unfold
This commit is contained in:
i-robot 2023-03-28 07:55:13 +00:00 committed by Gitee
parent 48efbd3531 6a05037f30
commit 4922f33a52
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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

@ -6203,7 +6203,7 @@ def unfold(input, kernel_size, dilation=1, padding=0, stride=1):
>>> x = Tensor(np.random.rand(4, 4, 32, 32), mindspore.float64)
>>> output = ops.unfold(x, kernel_size=3, dilation=1, stride=1)
>>> print(output.shape)
(4, 36, 30, 30)
(4, 4, 9, 900)
"""
kernel_size = _check_unfold_params(kernel_size, "kernel_size", [1, 2])
dilation = _check_unfold_params(dilation, "dilation", [1, 2])