#292 add some funcs

Merged
zoulq merged 92 commits from tanh01 into master 1 year ago
kcl commented 1 year ago
zoulq reviewed 1 year ago
ms_adapter/pytorch/nn/functional.py
@@ -570,6 +572,78 @@ def cross_entropy(input, target, weight=None, size_average=None, ignore_index=-1
result = ms.ops.cross_entropy(input, target, weight, ignore_index, reduction, label_smoothing)
return cast_to_adapter_tensor(result)

def ctc_loss(log_probs, targets, input_lengths, target_lengths, blank=0, reduction='mean', zero_infinity=False):
zoulq commented 1 year ago
新增接口都需要在上面列表中导出
zoulq reviewed 1 year ago
@@ -573,0 +634,4 @@
rlt = ms.ops.multilabel_soft_margin_loss(input, target, weight, reduction)
return cast_to_adapter_tensor(rlt)

# def multi_margin_loss(input, target, p=1, margin=1, weight=None, size_average=None, reduce=None, reduction='mean'):
zoulq commented 1 year ago
这个接口注释掉是因为什么功能不支持?
kcl commented 1 year ago
cpu不支持,本地没测
zoulq commented 1 year ago
mindspore.ops.multi_margin_loss---是否可以用这个接口,已测试该接口也可在GPU上支持
kcl commented 1 year ago
看错了,这个是重复开发,已经删了。是MultiLabelMarginLoss不支持CPU
zoulq reviewed 1 year ago
@@ -1938,0 +2120,4 @@

def unfold(input, kernel_size, dilation=1, padding=0, stride=1):
input_ms = cast_to_ms_tensor(input)
output = ms.ops.unfold(input_ms, kernel_size, dilation, padding, stride)
zoulq commented 1 year ago
ms.ops.unfold当前资料显示不支持GPU平台,可以标注一下todo
zoulq reviewed 1 year ago
@@ -1938,0 +2136,4 @@
else:
shape_tmp = kernel_size[0] * kernel_size[1]
input_ms = input_ms.reshape(shape[0], -1, shape_tmp, shape[2])
output = ms.ops.fold(input_ms, ms.Tensor(output_size), kernel_size, dilation, padding, stride)
zoulq commented 1 year ago
ms.ops.fold目前不支持Ascend平台,需要标注一下
zoulq reviewed 1 year ago
zoulq reviewed 1 year ago
testing/ut/pytorch/nn/functional/test_loss.py
@@ -0,0 +31,4 @@
assert np.allclose(result_ms.asnumpy(), result_torch.numpy())
assert result_ms.shape == result_torch.shape

def gaussian_nll_loss():
zoulq commented 1 year ago
测试用例的名字必须要以test_开头,才能被pytest执行
zoulq merged commit 1411897dda into master 1 year ago
The pull request has been merged as 1411897dda.
Sign in to join this conversation.
No reviewers
No Label
No Milestone
No Assignees
2 Participants
Notifications
Due Date

No due date set.

Dependencies

This pull request currently doesn't have any dependencies.

Loading…
There is no content yet.