p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!345 modify annotation of depthwiseconv2d 

Merge pull request !345 from wangnan39/modify_annotation_of_depthwise_conv
This commit is contained in:
mindspore-ci-bot 2020-04-15 17:27:21 +08:00 committed by Gitee
parent 6d98284103 4a292d48f9
commit 3e3998adf0
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/ops/operations/nn_ops.py
View File

@ -679,7 +679,7 @@ class DepthwiseConv2dNative(PrimitiveWithInfer):
Inputs:
- **input** (Tensor) - Tensor of shape :math:`(N, C_{in}, H_{in}, W_{in})`.
- **weight** (Tensor) - Set size of kernel is :math:`(K_1, K_2)`, then the shape is
:math:`(channel_{multiplier}, C_{in}, K_1, K_2)`.
:math:`(\text{channel_multiplier}, C_{in}, K_1, K_2)`.
Outputs:
Tensor of shape :math:`(N, C_{in} * \text{channel_multiplier}, H_{out}, W_{out})`.