mindspore 2.0.0a0__cp39-cp39-win_amd64.whl → 2.0.0rc1__cp39-cp39-win_amd64.whl

mindspore/ops/functional.py

@@ -19,17 +19,15 @@
 from mindspore.common._register_for_tensor import tensor_operator_registry
 from mindspore.ops import _constants
 from mindspore.ops.function import *
-from mindspore.ops.function.sparse_func import sparse_add
-from mindspore.ops.function.sparse_func import csr_mm
-from mindspore.ops.function.array_func import narrow
+from mindspore.ops.function.array_func import narrow, flatten
 from mindspore.ops import operations as P
 from mindspore.ops.primitive import Primitive
-from mindspore.ops.operations import _grad_ops, _csr_ops, _inner_ops, linalg_ops
+from mindspore.ops.operations import _grad_ops, _csr_ops, _inner_ops, linalg_ops, _scalar_ops, _sequence_ops
 from mindspore.ops.operations.math_ops import Median
-from mindspore.ops.operations.array_ops import UniqueConsecutive, Triu
+from mindspore.ops.operations.array_ops import UniqueConsecutive
 from mindspore.ops.operations.nn_ops import AdaptiveMaxPool2D
 from mindspore.ops.operations.math_ops import Roll
-from mindspore.ops.composite.array_ops import repeat_interleave
+from mindspore.ops.composite.math_ops import mm, dot
 
 typeof = Primitive('typeof')
 hastype = Primitive('hastype')
@@ -43,19 +41,32 @@ reduce_sum = P.ReduceSum()
 reduce_max = P.ReduceMax()
 reduce_min = P.ReduceMin()
 reduce_mean = P.ReduceMean()
-sort = P.Sort()
 tensor_range = P.Range()
 tensor_scatter_update = P.TensorScatterUpdate()
 scatter_nd_update = P.ScatterNdUpdate()
 mixed_precision_cast = _inner_ops.MixedPrecisionCast()
-partial = P.Partial()
-# depend: mount a node to another node
-depend = P.Depend()
-identity = P.identity()
 
 # Dynamic shape
 is_sequence_value_unknown = Primitive("IsShapeUnKnown")
 is_sequence_shape_unknown = Primitive("IsDimUnKnown")
+is_dynamic_sequence_element_unknown = Primitive("IsElementUnknown")
+
+partial = P.Partial()
+# depend: mount a node to another node
+depend = P.Depend()
+identity = P.identity()
+# tuple/list/scalar ops
+scalar_div = _scalar_ops.ScalarDiv()
+scalar_mod = _scalar_ops.ScalarMod()
+scalar_add = _scalar_ops.ScalarAdd()
+scalar_mul = _scalar_ops.ScalarMul()
+scalar_sub = _scalar_ops.ScalarSub()
+scalar_gt = _scalar_ops.scalar_gt()
+scalar_ge = _scalar_ops.scalar_ge()
+scalar_le = _scalar_ops.scalar_le()
+scalar_lt = _scalar_ops.scalar_lt()
+scalar_eq = _scalar_ops.scalar_eq()
+scalar_floordiv = _scalar_ops.ScalarFloordiv()
 
 tuple_setitem = Primitive('tuple_setitem')
 tuple_getitem = Primitive(_constants.kTupleGetItem)
@@ -74,22 +85,11 @@ make_list = Primitive('make_list')
 make_slice = Primitive('make_slice')
 tuple_equal = Primitive("tuple_equal")
 list_equal = Primitive("list_equal")
-scalar_add = Primitive(_constants.kScalarAdd)
-scalar_mul = Primitive(_constants.kScalarMul)
-scalar_sub = Primitive(_constants.kScalarSub)
-scalar_div = Primitive(_constants.kScalarDiv)
-scalar_floordiv = Primitive(_constants.kScalarFloordiv)
 scalar_log = Primitive('scalar_log')
 scalar_pow = Primitive(_constants.kScalarPow)
-scalar_gt = Primitive('scalar_gt')
-scalar_ge = Primitive('scalar_ge')
-scalar_le = Primitive('scalar_le')
-scalar_lt = Primitive('scalar_lt')
-scalar_eq = Primitive('scalar_eq')
 scalar_ne = Primitive('scalar_ne')
 scalar_uadd = Primitive(_constants.kScalarUadd)
 scalar_usub = Primitive(_constants.kScalarUsub)
-scalar_mod = Primitive(_constants.kScalarMod)
 string_eq = Primitive('string_eq')
 string_concat = Primitive('string_concat')
 bool_not = Primitive("bool_not")
@@ -103,8 +103,6 @@ in_dict = Primitive("in_dict")
 not_in_dict = Primitive("not_in_dict")
 broadcast_gradient_args = Primitive('BroadcastGradientArgs')
 array_reduce = Primitive('array_reduce')
-zeros = P.Zeros()
-zeros_like = P.ZerosLike()
 distribute = Primitive('distribute')
 embed = Primitive('embed')
 ref_to_embed = _grad_ops.RefToEmbed()
@@ -119,7 +117,7 @@ switch_layer = Primitive('switch_layer')
 # for sum bprop
 reduced_shape = Primitive("reduced_shape")
 # shape_mul:input must be shape multiply elements in tuple(shape)
-shape_mul = Primitive("shape_mul")
+shape_mul = _sequence_ops.shape_mul()
 
 tensor_operator_registry.register('add', P.Add)
 tensor_operator_registry.register('addr', addr)
@@ -131,16 +129,30 @@ tensor_operator_registry.register('any', P.ReduceAny)
 tensor_operator_registry.register('atan2', atan2)
 tensor_operator_registry.register('abs', P.Abs)
 tensor_operator_registry.register('baddbmm', baddbmm)
+tensor_operator_registry.register('geqrf', geqrf)
+tensor_operator_registry.register('histc', histc)
+tensor_operator_registry.register('real', real)
+tensor_operator_registry.register('reciprocal', reciprocal)
+tensor_operator_registry.register('rsqrt', rsqrt)
+tensor_operator_registry.register('bincount', bincount)
+tensor_operator_registry.register('slogdet', slogdet)
+tensor_operator_registry.register('trace', trace)
+tensor_operator_registry.register('tril', tril)
+tensor_operator_registry.register('chunk', chunk)
 tensor_operator_registry.register('sqrt', sqrt)
 tensor_operator_registry.register('square', square)
 tensor_operator_registry.register('sub', sub)
-tensor_operator_registry.register('triu', Triu)
+tensor_operator_registry.register('triu', triu)
 tensor_operator_registry.register('tan', P.Tan)
+tensor_operator_registry.register('t', t)
 tensor_operator_registry.register('acos', acos)
 tensor_operator_registry.register('cos', cos)
 tensor_operator_registry.register('acosh', acosh)
 tensor_operator_registry.register('cosh', P.Cosh)
+tensor_operator_registry.register('cov', cov)
 tensor_operator_registry.register('asin', asin)
+tensor_operator_registry.register('sin', sin)
+tensor_operator_registry.register('sinc', sinc)
 tensor_operator_registry.register('pow', P.Pow)
 tensor_operator_registry.register('negative', neg)
 tensor_operator_registry.register('amin', amin)
@@ -152,10 +164,11 @@ tensor_operator_registry.register('reshape', P.Reshape)
 tensor_operator_registry.register('reverse', P.ReverseV2)
 tensor_operator_registry.register('reverse_sequence', P.ReverseSequence)
 tensor_operator_registry.register('xlogy', P.Xlogy)
-tensor_operator_registry.register('flatten', P.Flatten)
+tensor_operator_registry.register('flatten', flatten)
 tensor_operator_registry.register('transpose', P.Transpose)
 tensor_operator_registry.register('broadcast_to', P.BroadcastTo)
 tensor_operator_registry.register('matmul', matmul)
+tensor_operator_registry.register('inner', inner)
 tensor_operator_registry.register('xdivy', P.Xdivy)
 tensor_operator_registry.register('argmax', P.Argmax)
 tensor_operator_registry.register('argmin', P.Argmin)
@@ -168,28 +181,41 @@ tensor_operator_registry.register('positive', positive)
 tensor_operator_registry.register('permute', permute)
 tensor_operator_registry.register('remainder', remainder)
 tensor_operator_registry.register('index_fill', index_fill)
+tensor_operator_registry.register('index_select', index_select)
 tensor_operator_registry.register('flip', flip)
 tensor_operator_registry.register('fliplr', fliplr)
 tensor_operator_registry.register('flipud', flipud)
+tensor_operator_registry.register('float_power', float_power)
+tensor_operator_registry.register('fmod', fmod)
 tensor_operator_registry.register('is_floating_point', is_floating_point)
 tensor_operator_registry.register('bitwise_and', bitwise_and)
 tensor_operator_registry.register('bitwise_or', bitwise_or)
 tensor_operator_registry.register('bitwise_xor', bitwise_xor)
+tensor_operator_registry.register('bitwise_left_shift', bitwise_left_shift)
+tensor_operator_registry.register('bitwise_right_shift', bitwise_right_shift)
 tensor_operator_registry.register('ger', ger)
 tensor_operator_registry.register('reduce_max', P.ReduceMax)
 tensor_operator_registry.register('reduce_min', P.ReduceMin)
 tensor_operator_registry.register('random_categorical', random_categorical)
 tensor_operator_registry.register('mirror_pad', P.MirrorPad)
 tensor_operator_registry.register('minimum', P.Minimum)
-tensor_operator_registry.register('matrix_determinant', matrix_determinant)
+tensor_operator_registry.register('matrix_power', matrix_power)
+tensor_operator_registry.register('det', det)
+tensor_operator_registry.register('dot', dot)
 tensor_operator_registry.register('log1p', log1p)
+tensor_operator_registry.register('logdet', logdet)
 tensor_operator_registry.register('log_matrix_determinant', log_matrix_determinant)
+tensor_operator_registry.register('matrix_determinant', matrix_determinant)
 tensor_operator_registry.register('ceil', P.Ceil)
 tensor_operator_registry.register('fill', P.Fill)
 tensor_operator_registry.register('tile', P.Tile)
 tensor_operator_registry.register('logit', logit)
 tensor_operator_registry.register('sum', P.ReduceSum)
-tensor_operator_registry.register('split', P.Split)
+tensor_operator_registry.register('split', split)
+tensor_operator_registry.register('tensor_split', tensor_split)
+tensor_operator_registry.register('vsplit', vsplit)
+tensor_operator_registry.register('hsplit', hsplit)
+tensor_operator_registry.register('dsplit', dsplit)
 tensor_operator_registry.register('select', P.Select)
 tensor_operator_registry.register('zeros_like', P.ZerosLike)
 tensor_operator_registry.register('scalar_to_tensor', scalar_to_tensor)
@@ -199,23 +225,29 @@ tensor_operator_registry.register('masked_select', masked_select)
 tensor_operator_registry.register('nonzero', nonzero)
 tensor_operator_registry.register('i0', i0)
 tensor_operator_registry.register('isclose', isclose)
+tensor_operator_registry.register('isneginf', isneginf)
+tensor_operator_registry.register('isposinf', isposinf)
+tensor_operator_registry.register('isreal', isreal)
 tensor_operator_registry.register('inv', inv)
+tensor_operator_registry.register('digamma', digamma)
+tensor_operator_registry.register('lgamma', lgamma)
 tensor_operator_registry.register('logaddexp', logaddexp)
 tensor_operator_registry.register('logaddexp2', logaddexp2)
 tensor_operator_registry.register('logsumexp', logsumexp)
+tensor_operator_registry.register('inverse', inverse)
 tensor_operator_registry.register('invert', invert)
 tensor_operator_registry.register('hardshrink', P.HShrink)
 tensor_operator_registry.register('heaviside', heaviside)
 tensor_operator_registry.register('hypot', hypot)
-tensor_operator_registry.register('soft_shrink', P.SoftShrink)
+tensor_operator_registry.register('soft_shrink', soft_shrink)
 tensor_operator_registry.register('svd', linalg_ops.Svd)
 tensor_operator_registry.register('diag', P.Diag)
+tensor_operator_registry.register('diagflat', diagflat)
 tensor_operator_registry.register('unique_consecutive', UniqueConsecutive)
 tensor_operator_registry.register('unique_with_pad', P.UniqueWithPad)
-tensor_operator_registry.register('inplace_update', P.InplaceUpdate)
+tensor_operator_registry.register('inplace_update', P.InplaceUpdateV2)
 tensor_operator_registry.register('col2im', col2im)
 tensor_operator_registry.register('standard_laplace', P.StandardLaplace)
-tensor_operator_registry.register('split', P.Split)
 tensor_operator_registry.register('erf', P.Erf)
 tensor_operator_registry.register('erfc', P.Erfc)
 tensor_operator_registry.register('standard_normal', P.StandardNormal)
@@ -228,8 +260,10 @@ tensor_operator_registry.register('addmm', addmm)
 tensor_operator_registry.register('addmv', addmv)
 tensor_operator_registry.register('adjoint', adjoint)
 tensor_operator_registry.register('asinh', asinh)
+tensor_operator_registry.register('arcsinh', arcsinh)
 tensor_operator_registry.register('atan', atan)
 tensor_operator_registry.register('atanh', atanh)
+tensor_operator_registry.register('arctanh', arctanh)
 tensor_operator_registry.register('bmm', bmm)
 tensor_operator_registry.register('conj', conj)
 tensor_operator_registry.register('cross', cross)
@@ -237,6 +271,7 @@ tensor_operator_registry.register('erfinv', erfinv)
 tensor_operator_registry.register('less_equal', less_equal)
 tensor_operator_registry.register('lcm', lcm)
 tensor_operator_registry.register('ldexp', ldexp)
+tensor_operator_registry.register('clamp', clamp)
 tensor_operator_registry.register('fold', fold)
 tensor_operator_registry.register('unfold', unfold)
 tensor_operator_registry.register('index_add', index_add)
@@ -244,6 +279,8 @@ tensor_operator_registry.register('greater', greater)
 tensor_operator_registry.register('greater_equal', greater_equal)
 tensor_operator_registry.register('igamma', igamma)
 tensor_operator_registry.register('igammac', igammac)
+tensor_operator_registry.register('nextafter', nextafter)
+tensor_operator_registry.register('qr', qr)
 # ms cannot support Tensor(True) compare
 tensor_operator_registry.register('__eq__', equal)
 tensor_operator_registry.register('__ne__', not_equal)
@@ -262,9 +299,9 @@ tensor_operator_registry.register('expand_dims', expand_dims)
 # support GE backend for no compare operators
 tensor_operator_registry.register('cast', cast)
 tensor_operator_registry.register('shape_mul', shape_mul)
+tensor_operator_registry.register('concatenate', P.Concat)
 tensor_operator_registry.register('fill', fill)
 tensor_operator_registry.register('fills', fills)
-tensor_operator_registry.register('concatenate', P.Concat)
 tensor_operator_registry.register('eye', eye)
 tensor_operator_registry.register('reduce_sum', reduce_sum)
 tensor_operator_registry.register('tensor_slice', tensor_slice)
@@ -296,11 +333,18 @@ tensor_operator_registry.register('dense_to_sparse_coo', dense_to_sparse_coo)
 tensor_operator_registry.register('csr_to_dense', csr_to_dense)
 tensor_operator_registry.register('narrow', narrow)
 tensor_operator_registry.register('sort', sort)
+tensor_operator_registry.register('argsort', argsort)
+tensor_operator_registry.register('msort', msort)
+tensor_operator_registry.register('mm', mm)
+tensor_operator_registry.register('nan_to_num', nan_to_num)
+tensor_operator_registry.register('nansum', nansum)
 tensor_operator_registry.register('csr_to_coo', csr_to_coo)
 tensor_operator_registry.register('zeros', zeros)
+tensor_operator_registry.register('ones', ones)
 tensor_operator_registry.register('unsorted_segment_min', unsorted_segment_min)
 tensor_operator_registry.register('unsorted_segment_max', unsorted_segment_max)
 tensor_operator_registry.register('unsorted_segment_prod', unsorted_segment_prod)
+tensor_operator_registry.register('scatter', scatter)
 tensor_operator_registry.register('tensor_scatter_update', tensor_scatter_update)
 tensor_operator_registry.register('tensor_scatter_mul', tensor_scatter_mul)
 tensor_operator_registry.register('tensor_scatter_div', tensor_scatter_div)
@@ -311,15 +355,16 @@ tensor_operator_registry.register('tensor_scatter_add', tensor_scatter_add)
 tensor_operator_registry.register('bernoulli', bernoulli)
 tensor_operator_registry.register('poisson', P.Poisson)
 tensor_operator_registry.register('randperm', P.Randperm)
-tensor_operator_registry.register('multinomial', P.Multinomial)
+tensor_operator_registry.register('multinomial', multinomial)
 tensor_operator_registry.register('norm', norm)
 tensor_operator_registry.register('renorm', renorm)
 tensor_operator_registry.register('adaptive_max_pool2d', AdaptiveMaxPool2D)
 tensor_operator_registry.register('coalesce', coalesce)
 tensor_operator_registry.register('argmax_with_value', max)
 tensor_operator_registry.register('argmin_with_value', min)
-tensor_operator_registry.register('coo_add', sparse_add)
-tensor_operator_registry.register('top_k', P.TopK)
+tensor_operator_registry.register('argwhere', argwhere)
+tensor_operator_registry.register('coo_add', coo_add)
+tensor_operator_registry.register('topk', topk)
 tensor_operator_registry.register('isfinite', P.IsFinite)
 tensor_operator_registry.register('to', P.Cast)
 tensor_operator_registry.register('bool', P.Cast)
@@ -331,11 +376,14 @@ tensor_operator_registry.register('cholesky', P.Cholesky)
 tensor_operator_registry.register('cholesky_inverse', P.CholeskyInverse)
 tensor_operator_registry.register('expand', expand)
 tensor_operator_registry.register('cumprod', cumprod)
+tensor_operator_registry.register('diff', diff)
 tensor_operator_registry.register('div', div)
 tensor_operator_registry.register('equal', equal)
 tensor_operator_registry.register('expm1', expm1)
+tensor_operator_registry.register('frac', frac)
 tensor_operator_registry.register('isinf', isinf)
 tensor_operator_registry.register('isnan', isnan)
+tensor_operator_registry.register('is_complex', is_complex)
 tensor_operator_registry.register('le', le)
 tensor_operator_registry.register('less', less)
 tensor_operator_registry.register('logical_and', logical_and)
@@ -345,13 +393,21 @@ tensor_operator_registry.register('logical_xor', logical_xor)
 tensor_operator_registry.register('lstsq', lstsq)
 tensor_operator_registry.register('mvlgamma', mvlgamma)
 tensor_operator_registry.register('maximum', maximum)
+tensor_operator_registry.register('max', max)
+tensor_operator_registry.register('min', min)
 tensor_operator_registry.register('mul', mul)
 tensor_operator_registry.register('multiply', multiply)
+tensor_operator_registry.register('moveaxis', moveaxis)
+tensor_operator_registry.register('movedim', movedim)
 tensor_operator_registry.register('neg', neg)
 tensor_operator_registry.register('ne', ne)
+tensor_operator_registry.register('not_equal', not_equal)
+tensor_operator_registry.register('sgn', sgn)
+tensor_operator_registry.register('sign', sign)
+tensor_operator_registry.register('signbit', signbit)
 tensor_operator_registry.register('sinh', sinh)
-tensor_operator_registry.register('sort', P.Sort)
 tensor_operator_registry.register('trunc', trunc)
+tensor_operator_registry.register('where', where)
 tensor_operator_registry.register('imag', imag)
 tensor_operator_registry.register('repeat_interleave', repeat_interleave)
 tensor_operator_registry.register('rad2deg', rad2deg)
@@ -359,6 +415,10 @@ tensor_operator_registry.register('deg2rad', deg2rad)
 tensor_operator_registry.register('copysign', copysign)
 tensor_operator_registry.register('roll', Roll)
 tensor_operator_registry.register('rot90', rot90)
+tensor_operator_registry.register('swapaxes', swapaxes)
+tensor_operator_registry.register('swapdims', swapdims)
+tensor_operator_registry.register('repeat_elements', repeat_elements)
+tensor_operator_registry.register('top_k', P.TopK)
 
 __all__ = [name for name in dir() if name[0] != "_"]
 __all__.remove('Primitive')