From a4e5b6bd407602e9b13a728580184282b8efd3cf Mon Sep 17 00:00:00 2001 From: lilei Date: Mon, 9 Nov 2020 17:09:15 +0800 Subject: [PATCH] modify guide for website --- mindspore/nn/layer/basic.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore/nn/layer/basic.py b/mindspore/nn/layer/basic.py index 1195435bd1..ee33c5144b 100644 --- a/mindspore/nn/layer/basic.py +++ b/mindspore/nn/layer/basic.py @@ -538,11 +538,11 @@ class Unfold(Cell): and the shape is [out_batch, out_depth, out_row, out_col], the out_batch is the same as the in_batch. Examples: - >>> net = Unfold(ksizes=[1, 2, 2, 1], strides=[1, 1, 1, 1], rates=[1, 1, 1, 1]) - >>> image = Tensor(np.ones([1, 1, 3, 3]), dtype=mstype.float16) + >>> net = Unfold(ksizes=[1, 2, 2, 1], strides=[1, 2, 2, 1], rates=[1, 2, 2, 1]) + >>> image = Tensor(np.ones([2, 3, 6, 6]), dtype=mstype.float16) >>> net(image) - Tensor ([[[[1, 1] [1, 1]] [[1, 1], [1, 1]] [[1, 1] [1, 1]], [[1, 1], [1, 1]]]], - shape=(1, 4, 2, 2), dtype=mstype.float16) + Tensor ([[[[1, 1] [1, 1]] [[1, 1], [1, 1]] [[1, 1] [1, 1]], [[1, 1] [1, 1]], [[1, 1] [1, 1]], + [[1, 1], [1, 1]]]], shape=(2, 12, 2, 2), dtype=mstype.float16) """ def __init__(self, ksizes, strides, rates, padding="valid"):