@@ -31,12 +31,12 @@ def _ms_max(*args, keep_dims=False, **kwargs):
3131 'Mindspore version to GitHub issues.' )
3232
3333_norm_signature = inspect .signature (mindspore .ops .norm )
34- if 'axis' in _norm_signature .parameters :
35- def _ms_norm (* args , axis = None , ** kwargs ):
36- return mindspore .ops .norm (* args , axis = axis , ** kwargs )
37- elif 'dim' in _norm_signature .parameters :
38- def _ms_norm (* args , axis = None , ** kwargs ):
39- return mindspore .ops .norm (* args , dim = axis , ** kwargs )
34+ if 'axis' in _norm_signature .parameters and 'p' in _norm_signature . parameters :
35+ def _ms_norm (* args , p = None , axis = None , ** kwargs ):
36+ return mindspore .ops .norm (* args , p = p , axis = axis , ** kwargs )
37+ elif 'dim' in _norm_signature .parameters and 'ord' in _norm_signature . parameters :
38+ def _ms_norm (* args , p = None , axis = None , ** kwargs ):
39+ return mindspore .ops .norm (* args , ord = p , dim = axis , ** kwargs )
4040else :
4141 raise ValueError ('Mindspore function mindspore.ops.norm has unsupported signature. It is likely you are '
4242 'working with a new Mindspore version which breaks backward compatibility. Please report your '
0 commit comments