diff --git a/mindspore/nn/layer/conv.py b/mindspore/nn/layer/conv.py index 9563bae1f1d..3ccab1a300f 100644 --- a/mindspore/nn/layer/conv.py +++ b/mindspore/nn/layer/conv.py @@ -482,9 +482,9 @@ class Conv3d(_Conv): 3D convolution layer. Applies a 3D convolution over an input tensor which is typically of shape - For input shape :math:`(N, C_{in}, D_{in}, H_{in}, W_{in})` and output shape - :math:`(N, C_{out}, D_{out}, H_{out}, W_{out})`. where :math:`N` is batch size. :math:`C` is channel number. - the formula is defined as: + :math:`(N, C_{in}, D_{in}, H_{in}, W_{in})` and output shape + :math:`(N, C_{out}, D_{out}, H_{out}, W_{out})`. where :math:`N` is batch size. :math:`C` is channel number. + the formula is defined as: .. math:: @@ -731,10 +731,10 @@ class Conv3dTranspose(_Conv): Supported Platforms: ``Ascend`` - Raise: + Raises: TypeError: If `in_channels`, `out_channels` or `group` is not an int. TypeError: If `kernel_size`, `stride`, `padding` , `dilation` or `output_padding` - is neither an int not a tuple of three. + is neither an int not a tuple of three. TypeError: If input data type is not float16 or float32. ValueError: If `in_channels`, `out_channels`, `kernel_size`, `stride` or `dilation` is less than 1. ValueError: If `padding` is less than 0. diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index 5cff0ece382..b836df973bd 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -7693,9 +7693,9 @@ class Conv3D(PrimitiveWithInfer): 3D convolution layer. Applies a 3D convolution over an input tensor which is typically of shape - For input shape :math:`(N, C_{in}, D_{in}, H_{in}, W_{in})` and output shape - :math:`(N, C_{out}, D_{out}, H_{out}, W_{out})`. where :math:`N` is batch size. :math:`C` is channel number. - the formula is defined as: + :math:`(N, C_{in}, D_{in}, H_{in}, W_{in})` and output shape + :math:`(N, C_{out}, D_{out}, H_{out}, W_{out})`. where :math:`N` is batch size. :math:`C` is channel number. + the formula is defined as: .. math:: @@ -7756,7 +7756,7 @@ class Conv3D(PrimitiveWithInfer): Inputs: - **input** (Tensor) - Tensor of shape :math:`(N, C_{in}, D_{in}, H_{in}, W_{in})`. - Currently input data type only support float16 and float32. + Currently input data type only support float16 and float32. - **weight** (Tensor) - Set size of kernel is :math:`(D_in, K_h, K_w)`, then the shape is :math:`(C_{out}, C_{in}, D_{in}, K_h, K_w)`. Currently weight data type only support float16 and float32. - **bias** (Tensor) - Tensor of shape :math:`C_{in}`. Currently, only support none. @@ -8121,7 +8121,7 @@ class Conv3DTranspose(PrimitiveWithInfer): Inputs: - **dout** (Tensor) - the gradients w.r.t the output of the convolution. The shape conforms to the default data_format :math:`(N, C_{in}, D_{out}, H_{out}, W_{out})`. Currently dout data type only support float16 - and float32. + and float32. - **weight** (Tensor) - Set size of kernel is :math:`(D_in, K_h, K_w)`, then the shape is :math:`(C_{in}//groups, C_{out}, D_{in}, K_h, K_w)`. Currently weight data type only support float16 and float32. @@ -8133,7 +8133,7 @@ class Conv3DTranspose(PrimitiveWithInfer): Supported Platforms: ``Ascend`` - Raise: + Raises: TypeError: If `in_channel`, `out_channel` or `group` is not an int. TypeError: If `kernel_size`, `stride`, `pad` , `dilation` or `output_padding` is neither an int not a tuple. ValueError: If `in_channel`, `out_channel`, `kernel_size`, `stride` or `dilation` is less than 1.