modify annotation of DepthwiseConv2d

pull/345/head
wangnan39@huawei.com 5 years ago
parent 6721541c9d
commit 4a292d48f9

@ -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})`.

Loading…
Cancel
Save