提交 3e3998ad 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!345 modify annotation of depthwiseconv2d

Merge pull request !345 from wangnan39/modify_annotation_of_depthwise_conv
......@@ -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})`.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册