Browse Source

!299 modify annotation of DepthwiseConv2d

Merge pull request !299 from wangnan39/modify_annotation_of_depthwise_conv
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
c06d2c6c2a
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/operations/nn_ops.py

+ 1
- 1
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:`(channel_{multiplier}, C_{in}, K_1, K_2)`.

Outputs:
Tensor of shape :math:`(N, C_{in} * \text{channel_multiplier}, H_{out}, W_{out})`.


Loading…
Cancel
Save