!49901 update ctcloss and unfold doc

Merge pull request !49901 from baihuawei/code_docs_ctcloss_and_undold
This commit is contained in:
i-robot 2023-03-08 01:49:59 +00:00 committed by Gitee
commit 4b7e743ce2
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions

View File

@ -2363,7 +2363,7 @@ class CTCLoss(LossBase):
>>> ctc_loss = CTCLoss(blank=0, reduction='none', zero_infinity=False)
>>> loss = ctc_loss(ms_input, target, input_lengths, target_lengths)
>>> print(loss)
[-25.794968]
-25.794968
"""
def __init__(self, blank=0, reduction='mean', zero_infinity=False):

View File

@ -6013,7 +6013,7 @@ def unfold(input, kernel_size, dilation=1, padding=0, stride=1):
ValueError: If `padding` value is less than zero.
Supported Platforms:
``Ascend`` ``CPU``
``Ascend`` ``GPU`` ``CPU``
Examples:
>>> x = Tensor(np.random.rand(4, 4, 32, 32), mindspore.float64)