mindspore 2.3.0__cp39-cp39-win_amd64.whl → 2.4.0__cp39-cp39-win_amd64.whl

mindspore/mint/__init__.py

@@ -15,45 +15,56 @@
 """mint module."""
 from __future__ import absolute_import
 import mindspore.ops as ops
-from mindspore.ops.extend import gather, conv2d, max, min
-from mindspore.ops.extend import array_func, math_func, nn_func
-from mindspore.mint.nn.functional import *
+from mindspore.common._register_for_tensor import tensor_operator_registry_for_mint
+from mindspore.common.tensor import Tensor
+from mindspore.ops.function.array_func import gather_ext as gather, max_ext as max, min_ext as min
+from mindspore.ops.function.nn_func import conv2d_ext as conv2d
+from mindspore.mint.nn.functional import sigmoid
 from mindspore.mint.nn import functional
 from mindspore.mint import linalg
-from mindspore.ops import erf, where, triu
+from mindspore.mint import special
+from mindspore.mint import distributed
+from mindspore.ops import erf, where
 from mindspore.ops.function.math_func import linspace_ext as linspace
-from mindspore.ops.function.array_func import full_ext as full
+from mindspore.ops.function.math_func import median_ext as median
 from mindspore.ops.function.array_func import ones_like_ext as ones_like
+from mindspore.ops.function.array_func import full_ext as full
 from mindspore.ops.function.array_func import zeros_like_ext as zeros_like
 from mindspore.ops.function.array_func import unique_ext as unique
+from mindspore.ops.function.array_func import chunk_ext as chunk
 from mindspore.ops.function.math_func import isclose
 from mindspore.ops.auto_generate import abs
 # 1
 from mindspore.ops.function.math_func import divide, div
 from mindspore.ops.auto_generate import topk_ext as topk
+from mindspore.ops.function.math_func import roll
 # 2
 from mindspore.ops.function.math_func import sin
 # 3
 from mindspore.ops.function.clip_func import clamp
 # 4
-
+from mindspore.ops.auto_generate import sinc
+from mindspore.ops.auto_generate import sinh
+from mindspore.ops.auto_generate import cosh
+from mindspore.ops.function.math_func import xlogy_ext as xlogy
 # 5
 from mindspore.ops.auto_generate import cumsum_ext as cumsum
 # 6
 from mindspore.ops.auto_generate import stack_ext as stack
 
 # 7
-
+from mindspore.ops.function.array_func import unsqueeze
 # 8
-
+from mindspore.ops.auto_generate import transpose_ext as transpose
 # 9
-
+from mindspore.ops.auto_generate import masked_select
+from mindspore.ops.function.math_func import cross
 # 10
 from mindspore.ops.function.math_func import ne
 # 11
 
 # 12
-
+from mindspore.ops.function.array_func import repeat_interleave_ext as repeat_interleave
 # 13
 from mindspore.ops.functional import flip
 # 14
@@ -122,13 +133,13 @@ from mindspore.ops.functional import cos
 # 45
 
 # 46
-
+from mindspore.ops.function.math_func import bitwise_and_ext as bitwise_and
 # 47
-
+from mindspore.ops.function.math_func import bitwise_or_ext as bitwise_or
 # 48
-
+from mindspore.ops.function.math_func import bitwise_xor_ext as bitwise_xor
 # 49
-
+from mindspore.ops.function.math_func import baddbmm_ext as baddbmm
 # 50
 from mindspore.ops.functional import tile
 # 51
@@ -142,7 +153,7 @@ from mindspore.ops.function.random_func import normal_ext as normal
 # 55
 
 # 56
-
+from mindspore.ops.function.math_func import norm_ext as norm
 # 57
 from mindspore.ops.functional import broadcast_to
 # 58
@@ -166,7 +177,7 @@ from mindspore.ops.functional import logical_not
 # 67
 from mindspore.ops.functional import logical_or
 # 68
-
+from mindspore.ops.functional import logical_xor
 # 69
 from mindspore.ops.functional import less_equal, le
 # 70
@@ -194,7 +205,7 @@ from mindspore.ops.function import arange_ext as arange
 # 81
 from mindspore.ops.auto_generate import index_select_ext as index_select
 # 82
-
+from mindspore.ops.auto_generate import cummin_ext as cummin
 # 83
 from mindspore.ops.function.array_func import narrow_ext as narrow
 # 84
@@ -204,9 +215,9 @@ from mindspore.mint import nn, optim
 # 86
 
 # 87
-
+from mindspore.ops.auto_generate import trunc
 # 88
-from mindspore.ops.function.array_func import chunk_ext as chunk
+
 # 89
 
 # 90
@@ -231,13 +242,82 @@ from mindspore.ops.function.math_func import tanh
 
 # 100
 
+# 101
+
+# 102
+
+# 103
+
+# 104
+
+# 105
+
+# 106
+
+# 107
+
+# 108
+
+# 109
+from mindspore.ops.auto_generate import argmin_ext as argmin
+# 110
+
+# 111
+
+# 112
+
+# 113
+
+# 114
+
+# 115
+
+# 116
+
+# 117
+
+# 118
+
+# 119
+
+# 120
+
+# 121
+
 # 122
 
+# 151
+from mindspore.ops.function.math_func import acos_ext as acos
+from mindspore.ops.function.math_func import arccos_ext as arccos
+# 152
+from mindspore.ops.function.math_func import acosh_ext as acosh
+from mindspore.ops.function.math_func import arccosh_ext as arccosh
+# 172
+from mindspore.ops.function.math_func import asin_ext as asin
+from mindspore.ops.function.math_func import arcsin_ext as arcsin
+# 173
+from mindspore.ops.function.math_func import asinh_ext as asinh
+from mindspore.ops.function.math_func import arcsinh_ext as arcsinh
+# 174
+from mindspore.ops.function.math_func import atan_ext as atan
+from mindspore.ops.function.math_func import arctan_ext as arctan
+# 175
+from mindspore.ops.function.math_func import atanh
+from mindspore.ops.function.math_func import arctanh
 # 176
 from mindspore.ops.function.math_func import atan2_ext as atan2
 from mindspore.ops.function.math_func import arctan2_ext as arctan2
 
+# 177
+from mindspore.ops.function.math_func import round
+
+# 182
+from mindspore.ops.function.math_func import bernoulli_ext as bernoulli
 
+# 204
+from mindspore.ops.auto_generate import erfc
+# 207
+from mindspore.ops.auto_generate import expm1
 # 208
 from mindspore.ops.function.array_func import eye
 from mindspore.ops.function.random_func import rand_ext as rand
@@ -246,9 +326,43 @@ from mindspore.ops.function.random_func import rand_like_ext as rand_like
 from mindspore.ops.auto_generate import floor
 # 231
 from mindspore.ops.function.math_func import inverse_ext as inverse
-
+# 244
+from mindspore.ops.auto_generate import log1p
+# 261
+from mindspore.ops.function.random_func import multinomial_ext as multinomial
+# 275
+from mindspore.ops.function.math_func import remainder_ext as remainder
 # 285
 from mindspore.ops.function.array_func import scatter_add_ext as scatter_add
+# 289
+from mindspore.ops.auto_generate import sign
+
+from mindspore.ops.auto_generate import select_ext as select
+
+# 301
+from mindspore.ops.function.math_func import tan
+
+# 303
+from mindspore.ops.auto_generate import trace_ext as trace
+
+from mindspore.ops.function.array_func import reshape
+
+from mindspore.ops.auto_generate import outer_ext as outer
+
+# 304
+from mindspore.ops.function.array_func import tril_ext as tril
+
+# 305
+from mindspore.ops import triu
+
+# 538
+from mindspore.ops.function.math_func import histc_ext as histc
+
+# 553
+from mindspore.ops.auto_generate import logaddexp_ext as logaddexp
+
+# 610
+from mindspore.ops.function.math_func import nan_to_num
 
 
 def add(input, other, *, alpha=1):
@@ -268,12 +382,12 @@ def add(input, other, *, alpha=1):
     Args:
         input (Union[Tensor, number.Number, bool]): The first input is a number.Number or
             a bool or a tensor whose data type is
-            `number <https://www.mindspore.cn/docs/en/master/api_python/mindspore.html#mindspore.dtype>`_ or
-            `bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore.html#mindspore.dtype>`_.
+            `number <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_ or
+            `bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_.
         other (Union[Tensor, number.Number, bool]): The second input, is a number.Number or
             a bool or a tensor whose data type is
-            `number <https://www.mindspore.cn/docs/en/master/api_python/mindspore.html#mindspore.dtype>`_ or
-            `bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore.html#mindspore.dtype>`_.
+            `number <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_ or
+            `bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_.
 
     Keyword Args:
         alpha (number.Number): A scaling factor applied to `other`, default 1.
@@ -429,7 +543,6 @@ def all(input, dim=None, keepdim=False):
     return ops.function.math_func.all(input, dim, keepdim)
 
 
-
 def cat(tensors, dim=0):
     r"""
     Connect input tensors along with the given dimension.
@@ -487,6 +600,104 @@ def cat(tensors, dim=0):
     return ops.auto_generate.cat(tensors, dim)
 
 
+def concat(tensors, dim=0):
+    r"""
+    .. warning::
+        This is an experimental API that is subject to change or deletion.
+
+    Alias of mint.cat().
+    """
+    return cat(tensors, dim)
+
+
+def cummax(input, dim):
+    r"""
+    Returns a tuple (values, indices) where `values` is the cumulative maximum value of input Tensor `input`
+    along the dimension `dim`, and `indices` is the index location of each maximum value.
+
+    .. math::
+        \begin{array}{ll} \\
+            y_{i} = \max(x_{1}, x_{2}, ... , x_{i})
+        \end{array}
+
+    Args:
+        input (Tensor): The input Tensor. Rank of `input` must be greater than 0.
+        dim (int): The dimension to do the operation over. The value of `dim` must be in the range
+            `[-input.ndim, input.ndim - 1]`.
+
+    Returns:
+        tuple [Tensor], tuple of 2 Tensors, containing the cumulative maximum of elements and the index.
+        The shape of each output tensor is the same as that of input `input`.
+
+    Raises:
+        TypeError: If `input` is not a Tensor.
+        TypeError: If `dim` is not an int.
+        ValueError: If `dim` is out the range of `[-input.ndim, input.ndim - 1]`.
+
+    .. note::
+        O2 mode is not supported in Ascend.
+
+    Supported Platforms:
+        ``Ascend``
+
+    Examples:
+        >>> import mindspore
+        >>> import numpy as np
+        >>> from mindspore import Tensor
+        >>> from mindspore import ops
+        >>> x = Tensor(np.array([[3, 4, 6, 10], [1, 6, 7, 9], [4, 3, 8, 7], [1, 3, 7, 9]]).astype(np.float32))
+        >>> output = mint.cummax(x, dim=0)
+        >>> print(output[0])
+        [[ 3.  4.  6. 10.]
+         [ 3.  6.  7. 10.]
+         [ 4.  6.  8. 10.]
+         [ 4.  6.  8. 10.]]
+        >>> print(output[1])
+        [[0 0 0 0]
+         [0 1 1 0]
+         [2 1 2 0]
+         [2 1 2 0]]
+    """
+    return ops.auto_generate.cummax(input, dim)
+
+
+def item(input):
+    r"""
+    Returns the value of this tensor as a standard Python number.
+
+    Note:
+        This only works for tensors with one element.
+
+    Args:
+        input (Tensor[Number]): The input tensor. The dtype of the tensor to be reduced is number.
+            :math:`(N, *)` where :math:`*` means, any number of additional dimensions.
+
+    Returns:
+        number.
+
+    Raises:
+        TypeError: If `input` is not a Tensor.
+        RuntimeError: If the number of `input` elements is not 1.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import mindspore
+        >>> import numpy as np
+        >>> from mindspore import Tensor, mint
+        >>> x = Tensor(np.array([1]).astype(np.float32))
+        >>> result = mint.item(x)
+        >>> print(result)
+        1.0
+    """
+    if not isinstance(input, Tensor):
+        raise TypeError(f"the input must be a Tensor, but got {type(input)}")
+    if input.size != 1:
+        raise RuntimeError("a Tensor with {} elements cannot be converted to Scalar".format(input.size))
+    return input.asnumpy().item()
+
+
 def mean(input, dim=None, keepdim=False, *, dtype=None):
     r"""
     Reduces all dimension of a tensor by averaging all elements in the dimension, by default.
@@ -818,12 +1029,12 @@ def sub(input, other, *, alpha=1):
     Args:
         input (Union[Tensor, number.Number, bool]): The first input is a number.Number or
             a bool or a tensor whose data type is
-            `number <https://www.mindspore.cn/docs/en/master/api_python/mindspore.html#mindspore.dtype>`_ or
-            `bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore.html#mindspore.dtype>`_.
+            `number <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_ or
+            `bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_.
         other (Union[Tensor, number.Number, bool]): The second input, is a number.Number or
             a bool or a tensor whose data type is
-            `number <https://www.mindspore.cn/docs/en/master/api_python/mindspore.html#mindspore.dtype>`_ or
-            `bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore.html#mindspore.dtype>`_.
+            `number <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_ or
+            `bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_.
 
     Keyword Args:
         alpha (number.Number): A scaling factor applied to `other`, default 1.
@@ -859,6 +1070,22 @@ def sub(input, other, *, alpha=1):
     return ops.auto_generate.sub_ext(input, other, alpha)
 
 
+def swapaxes(input, axis0, axis1):
+    '''
+    Interchange two axes of a tensor, alias for mint.transpose()
+
+    Examples:
+        >>> import numpy as np
+        >>> from mindspore import mint
+        >>> from mindspore import Tensor
+        >>> input = Tensor(np.ones((2,3,4), dtype=np.float32))
+        >>> output = mint.swapaxes(input, 0, 2)
+        >>> print(output.shape)
+        (4, 3, 2)
+    '''
+    return transpose(input, axis0, axis1)
+
+
 def zeros(size, *, dtype=None):
     """
     Creates a tensor filled with 0 with shape described by `size` and fills it with value 0 in type of `dtype`.
@@ -892,25 +1119,112 @@ def zeros(size, *, dtype=None):
     return ops.auto_generate.zeros(size, dtype)
 
 
+def fix(input):
+    """
+    Alias for :func:`mindspore.mint.trunc` .
+
+    For more details, see :func:`mindspore.mint.trunc` .
+
+    Supported Platforms:
+        ``Ascend``
+    """
+    return trunc(input)
+
+
+def scatter(input, dim, index, src):
+    """
+    Update the value in `src` to `input` according to the specified index.
+    For a 3-D tensor, the output will be:
+
+    .. code-block::
+
+        output[index[i][j][k]][j][k] = src[i][j][k]  # if dim == 0
+
+        output[i][index[i][j][k]][k] = src[i][j][k]  # if dim == 1
+
+        output[i][j][index[i][j][k]] = src[i][j][k]  # if dim == 2
+
+    .. note::
+        The backward is supported only for the case `src.shape == index.shape` when `src` is a tensor.
+
+    Args:
+        input (Tensor): The target tensor. The rank of `input` must be at least 1.
+        dim (int): Which axis to scatter. Accepted range is [-r, r) where r = rank(input).
+        index (Tensor): The index to do update operation whose data must be positive number with type of mindspore.int32
+            or mindspore.int64. Same rank as `input` . And accepted range is [-s, s) where s is the size along axis.
+        src (Tensor, float): The data doing the update operation with `input`. Can be a tensor with the same data type
+            as `input` or a float number to scatter.
+
+    Returns:
+        Tensor, has the same shape and type as `input` .
+
+    Raises:
+        TypeError: If `index` is neither int32 nor int64.
+        ValueError: If rank of any of `input` , `index` and `src` less than 1.
+        ValueError: If the rank of `src` is not equal to the rank of `input` .
+        TypeError: If the data type of `input` and `src` have different dtypes.
+        RuntimeError: If `index` has negative elements.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import numpy as np
+        >>> import mindspore as ms
+        >>> from mindspore import Tensor, mint
+        >>> input = Tensor(np.array([[1, 2, 3, 4, 5]]), dtype=ms.float32)
+        >>> src = Tensor(np.array([[8, 8]]), dtype=ms.float32)
+        >>> index = Tensor(np.array([[2, 4]]), dtype=ms.int64)
+        >>> out = mint.scatter(input=input, dim=1, index=index, src=src)
+        >>> print(out)
+        [[1. 2. 8. 4. 8.]]
+        >>> input = Tensor(np.zeros((5, 5)), dtype=ms.float32)
+        >>> src = Tensor(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]), dtype=ms.float32)
+        >>> index = Tensor(np.array([[0, 0, 0], [2, 2, 2], [4, 4, 4]]), dtype=ms.int64)
+        >>> out = mint.scatter(input=input, dim=0, index=index, src=src)
+        >>> print(out)
+        [[1. 2. 3. 0. 0.]
+        [0. 0. 0. 0. 0.]
+        [4. 5. 6. 0. 0.]
+        [0. 0. 0. 0. 0.]
+        [7. 8. 9. 0. 0.]]
+        >>> input = Tensor(np.zeros((5, 5)), dtype=ms.float32)
+        >>> src = Tensor(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]), dtype=ms.float32)
+        >>> index = Tensor(np.array([[0, 2, 4], [0, 2, 4], [0, 2, 4]]), dtype=ms.int64)
+        >>> out = mint.scatter(input=input, dim=1, index=index, src=src)
+        >>> print(out)
+        [[1. 0. 2. 0. 3.]
+        [4. 0. 5. 0. 6.]
+        [7. 0. 8. 0. 9.]
+        [0. 0. 0. 0. 0.]
+        [0. 0. 0. 0. 0.]]
+    """
+    return ops.function.array_func.scatter(input, dim, index, src)
+
+
 __all__ = [
+    'conv2d',
     'full',
     'ones_like',
     'zeros_like',
     'abs',
     'erf',
     'where',
-    'linspace',
     'isclose',
     # 1
     'div',
     'divide',
     'topk',
+    'roll',
     # 2
     'sin',
     # 3
     'clamp',
+    'xlogy',
     # 4
-
+    'sinc',
+    'sinh',
+    'cosh',
     # 5
     'cumsum',
     # 6
@@ -918,15 +1232,16 @@ __all__ = [
     # 7
     'zeros',
     # 8
-
+    'transpose',
+    'swapaxes',
     # 9
 
     # 10
     'ne',
     # 11
-
+    'unsqueeze',
     # 12
-
+    "repeat_interleave",
     # 13
     "flip",
     # 14
@@ -966,8 +1281,8 @@ __all__ = [
     # 30
     'searchsorted',
     # 31
-
-    # 32
+    'cummax',
+    'cummin',
     'sub',
     # 33
     'split',
@@ -994,15 +1309,17 @@ __all__ = [
     # 44
     'cos',
     # 45
-
+    'concat',
     # 46
-
+    'bitwise_and',
+    'bitwise_or',
+    'bitwise_xor',
     # 47
     'max',
     # 48
     'min',
     # 49
-
+    'baddbmm',
     # 50
     'tile',
     # 51
@@ -1014,9 +1331,9 @@ __all__ = [
     # 54
     'normal',
     # 55
-
+    'cross',
     # 56
-
+    'norm',
     # 57
     'broadcast_to',
     # 58
@@ -1040,7 +1357,7 @@ __all__ = [
     # 67
     'logical_or',
     # 68
-
+    'logical_xor',
     # 69
     'less_equal',
     'le',
@@ -1077,9 +1394,10 @@ __all__ = [
     'narrow',
     # 84
 
-    # 85
+    'masked_select',
 
     # 86
+    'select',
 
     # 87
 
@@ -1109,29 +1427,160 @@ __all__ = [
 
     # 100
 
+    # 101
+
+    # 102
+
+    # 103
+
+    # 104
+
+    # 105
+
+    # 106
+
+    # 107
+
+    # 108
+
+    # 109
+    'argmin',
+    # 110
+
+    # 111
+
+    # 112
+
+    # 113
+
+    # 114
+
+    # 115
+
+    # 116
+
+    # 117
+
+    # 118
+
+    # 119
+
+    # 120
+
+    # 121
+
+    # 122
+
+    # 151
+    'acos',
+    'arccos',
+    # 152
+    'acosh',
+    'arccosh',
+    # 153
+
+    # 154
+
+    # 155
+
+    # 156
+
+    # 157
+    'scatter',
+    # 172
+    'asin',
+    'arcsin',
+    # 173
+    'asinh',
+    'arcsinh',
+    # 174
+    'atan',
+    'arctan',
+    # 175
+    'atanh',
+    'arctanh',
     # 176
     'atan2',
     'arctan2',
 
+    # 177
+    'round',
+
+    # 182
+    'bernoulli',
+
+    # 207
+    'expm1',
+    # 204
+    'erfc',
     # 208
     'eye',
+
+    # 256
+    'median',
     'rand',
     'rand_like',
     # 210
     'floor',
     # 231
     'inverse',
+    # 244
+    'log1p',
+    # 261
+    'multinomial',
+    # 275
+    'remainder',
     # 285
     'scatter_add',
+    # 289
+    'sign',
+    # 301
+    'tan',
+    # 303
+    'trace',
+    'reshape',
+    'outer',
     # 304
+    'tril',
 
     # 305
     'triu',
+
+    # 538
+    'histc',
+
+    # 553
+    'logaddexp',
+
+    # 610
+    'nan_to_num',
 ]
-__all__.extend(array_func.__all__)
-__all__.extend(math_func.__all__)
-__all__.extend(nn_func.__all__)
+
+setattr(tensor_operator_registry_for_mint, 'add', add)
+setattr(tensor_operator_registry_for_mint, 'all', all)
+setattr(tensor_operator_registry_for_mint, 'any', any)
+setattr(tensor_operator_registry_for_mint, 'log', log)
+setattr(tensor_operator_registry_for_mint, 'ceil', ceil)
+setattr(tensor_operator_registry_for_mint, 'clamp', clamp)
+setattr(tensor_operator_registry_for_mint, 'cos', cos)
+setattr(tensor_operator_registry_for_mint, 'flatten', flatten)
+setattr(tensor_operator_registry_for_mint, 'item', item)
+setattr(tensor_operator_registry_for_mint, 'max', max)
+setattr(tensor_operator_registry_for_mint, 'mean', mean)
+setattr(tensor_operator_registry_for_mint, 'min', min)
+setattr(tensor_operator_registry_for_mint, 'repeat_interleave', repeat_interleave)
+setattr(tensor_operator_registry_for_mint, 'ne', ne)
+setattr(tensor_operator_registry_for_mint, 'round', round)
+setattr(tensor_operator_registry_for_mint, 'sin', sin)
+setattr(tensor_operator_registry_for_mint, 'split', split)
+setattr(tensor_operator_registry_for_mint, 'sqrt', sqrt)
+setattr(tensor_operator_registry_for_mint, 'square', square)
+setattr(tensor_operator_registry_for_mint, 'sub', sub)
+setattr(tensor_operator_registry_for_mint, 'sum', sum)
+
 __all__.extend(functional.__all__)
 __all__.extend(nn.__all__)
 __all__.extend(optim.__all__)
 __all__.extend(linalg.__all__)
+__all__.extend(special.__all__)
+__all__.extend(distributed.__all__)