mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!37796 fix dropout3d en doc description api 

Merge pull request !37796 from zhuzhongrui/code_docs_branch1
This commit is contained in:
i-robot 2022-07-11 11:46:20 +00:00 committed by Gitee
parent 7bb6983ea6 e5e3458754
commit a5a661e9eb
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mindspore/python/mindspore/nn/layer/basic.py
View File

@ -191,13 +191,6 @@ class Dropout2d(Cell):
Refer to :func:`mindspore.ops.dropout2d` for more details.
Raises:
TypeError: If `x` is not a Tensor.
TypeError: If dtype of `x` is not int8, int16, int32, int64, float16, float32 or float64.
TypeError: If the data type of `p` is not float.
ValueError: If `p` is out of the range `[0.0, 1.0]`.
ValueError: If `x` shape is not 4D.
Supported Platforms:
``Ascend`` ``GPU`` ``CPU``
@ -238,7 +231,7 @@ class Dropout3d(Cell):
During training, randomly zeroes some channels of the input tensor
with probability `p` from a Bernoulli distribution(For a 5-dimensional tensor with
a shape of :math:`NCDHW`, the channel feature map refers to a 3-dimensional feature
map with a shape of :math:'DHW').
map with a shape of :math:`DHW`).
For example, the :math:`j\_th` channel of the :math:`i\_th` sample in the batched input is a to-be-processed
`3D` tensor input[i,j].
@ -249,13 +242,6 @@ class Dropout3d(Cell):
Refer to :func:`mindspore.ops.dropout3d` for more details.
Raises:
TypeError: If `x` is not a Tensor.
TypeError: If dtype of `x` is not int8, int16, int32, int64, float16, float32 or float64.
TypeError: If the data type of `p` is not float.
ValueError: If `p` is out of the range `[0.0, 1.0]`.
ValueError: If `x` shape is not `5D`.
Supported Platforms:
``Ascend`` ``GPU`` ``CPU``