forked from mindspore-Ecosystem/mindspore
!49901 update ctcloss and unfold doc
Merge pull request !49901 from baihuawei/code_docs_ctcloss_and_undold
This commit is contained in:
commit
4b7e743ce2
|
@ -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):
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue