mindspore 2.4.0__cp311-cp311-macosx_10_15_x86_64.whl

mindspore/mint/__init__.py

@@ -0,0 +1,1586 @@
+# Copyright 2024 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+"""mint module."""
+from __future__ import absolute_import
+import mindspore.ops as ops
+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.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.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
+
+# 15
+from mindspore.ops.auto_generate import flatten_ext as flatten
+# 16
+from mindspore.ops.functional import matmul
+from mindspore.ops.auto_generate import bmm_ext as bmm
+# 17
+
+# 18
+from mindspore.ops.functional import sum
+# 19
+from mindspore.ops.functional import log
+# 20
+
+# 21
+from mindspore.ops.functional import mul
+# 22
+
+# 23
+
+# 24
+
+# 25
+from mindspore.ops.functional import greater, gt
+# 26
+from mindspore.ops.functional import eq
+# 27
+from mindspore.ops.functional import reciprocal
+# 28
+from mindspore.ops.functional import exp
+# 29
+
+# 30
+from mindspore.ops.functional import searchsorted
+# 31
+
+# 32
+
+# 33
+
+# 34
+
+# 35
+from mindspore.ops.functional import erfinv
+# 36
+
+# 37
+from mindspore.ops.function.array_func import nonzero
+# 38
+
+# 39
+
+# 40
+
+# 41
+
+# 42
+from mindspore.ops.function.math_func import argmax_ext as argmax
+# 43
+
+# 44
+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
+
+# 52
+
+# 53
+
+# 54
+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
+from mindspore.ops.function.math_func import greater_equal
+# 59
+from mindspore.ops.functional import square
+# 60
+
+# 61
+from mindspore.ops.functional import rsqrt
+# 62
+from mindspore.ops.functional import maximum
+# 63
+from mindspore.ops.functional import minimum
+# 64
+
+# 65
+from mindspore.ops.functional import logical_and
+# 66
+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
+from mindspore.ops.functional import negative, neg
+# 71
+from mindspore.ops.functional import isfinite
+# 72
+
+# 73
+from mindspore.ops.functional import ceil
+# 74
+from mindspore.ops.function.array_func import sort_ext as sort
+# 75
+from mindspore.ops.functional import less, lt
+# 76
+from mindspore.ops.functional import pow
+# 77
+
+# 78
+from mindspore.ops.function import arange_ext as arange
+# 79
+
+# 80
+
+# 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
+
+# 85
+from mindspore.mint import nn, optim
+# 86
+
+# 87
+from mindspore.ops.auto_generate import trunc
+# 88
+
+# 89
+
+# 90
+
+# 91
+
+# 92
+
+# 93
+
+# 94
+from mindspore.ops.function.math_func import tanh
+# 95
+
+# 96
+
+# 97
+
+# 98
+
+# 99
+
+# 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
+from mindspore.ops.function.random_func import rand_like_ext as rand_like
+# 210
+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):
+    r"""
+    Adds scaled other value to input Tensor.
+
+    .. math::
+
+        out_{i} = input_{i} + alpha \times other_{i}
+
+    Note:
+        - When the two inputs have different shapes,
+          they must be able to broadcast to a common shape.
+        - The two inputs and alpha comply with the implicit type conversion rules to make the data types
+          consistent.
+
+    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/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/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.
+
+    Returns:
+        Tensor with a shape that is the same as the broadcasted shape of the input `input` and `other`,
+        and the data type is the one with higher precision or higher digits among the two inputs and alpha.
+
+    Raises:
+        TypeError: If the type of `input`, `other`, or `alpha` is not one of the following: Tensor, number.Number, bool.
+        TypeError: If `alpha` is of type float but `input` and `other` are not of type float.
+        TypeError: If `alpha` is of type bool but `input` and `other` are not of type bool.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import numpy as np
+        >>> import mindspore
+        >>> from mindspore import Tensor
+        >>> from mindspore import mint
+        >>> x = Tensor(1, mindspore.int32)
+        >>> y = Tensor(np.array([4, 5, 6]).astype(np.float32))
+        >>> alpha = 0.5
+        >>> output = mint.add(x, y, alpha=alpha)
+        >>> print(output)
+        [3. 3.5 4.]
+        >>> # the data type of x is int32, the data type of y is float32,
+        >>> # alpha is a float, and the output is the data format of higher precision float32.
+        >>> print(output.dtype)
+        Float32
+    """
+    return ops.auto_generate.add_ext(input, other, alpha)
+
+
+def any(input, dim=None, keepdim=False):
+    r"""
+    Reduces a dimension of `input` by the "logical OR" of all elements in the dimension, by default. And also can
+    reduce a dimension of `input` along the `dim`. Determine whether the dimensions of the output and input are the
+    same by controlling `keepdim`.
+
+    Note:
+        The `dim` with tensor type is only used for compatibility with older versions and is not recommended.
+
+    Args:
+        input (Tensor): Input Tensor, has the shape :math:`(N, *)` where :math:`*` means,
+            any number of additional dimensions.
+        dim (Union[int, tuple(int), list(int), Tensor], optional): The dimensions to reduce.
+            Suppose the rank of `input` is r, `dim` must be in the range [-rank(input), rank(input)).
+            Default: ``None`` , all dimensions are reduced.
+        keepdim (bool, optional): If ``True`` , keep these reduced dimensions and the length is 1.
+            If ``False`` , don't keep these dimensions. Default : ``False`` .
+
+    Returns:
+        Tensor, the dtype is bool.
+
+        - If `dim` is ``None`` , and `keepdim` is ``False`` ,
+          the output is a 0-D Tensor representing the "logical OR" of all elements in the input Tensor.
+        - If `dim` is int, such as 2, and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_1, input_3, ..., input_R)`.
+        - If `dim` is tuple(int), such as (2, 3), and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_1, input_4, ..., input_R)`.
+        - If `dim` is 1-D Tensor, such as [2, 3], and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_1, input_4, ..., input_R)`.
+
+    Raises:
+        TypeError: If `keepdim` is not a bool.
+        TypeError: If `input` is not a Tensor.
+        TypeError: If `dim` is not one of the following: int, tuple, list or Tensor.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import numpy as np
+        >>> from mindspore import Tensor, mint
+        >>> x = Tensor(np.array([[True, False], [True, True]]))
+        >>> # case 1: Reduces a dimension by the "logical OR" of all elements in the dimension.
+        >>> output = mint.any(x, keepdim=True)
+        >>> print(output)
+        [[ True]]
+        >>> print(output.shape)
+        (1, 1)
+        >>> # case 2: Reduces a dimension along dim 0.
+        >>> output = mint.any(x, dim=0)
+        >>> print(output)
+        [ True True]
+        >>> # case 3: Reduces a dimension along dim 1.
+        >>> output = mint.any(x, dim=1)
+        >>> print(output)
+        [ True True]
+    """
+    return ops.functional.any(input, dim, keepdim)
+
+
+def all(input, dim=None, keepdim=False):
+    r"""
+    Reduces a dimension of `input` by the "logical AND" of all elements in the dimension, by default. And also can
+    reduce a dimension of `input` along the `dim`. Determine whether the dimensions of the output and input are the
+    same by controlling `keepdim`.
+
+    Note:
+        The `dim` with tensor type is only used for compatibility with older versions and is not recommended.
+
+    Args:
+        input (Tensor): Input Tensor, has the shape :math:`(N, *)` where :math:`*` means,
+            any number of additional dimensions.
+        dim (Union[int, tuple(int), list(int), Tensor], optional): The dimensions to reduce.
+            Suppose the rank of `input` is r, `dim` must be in the range [-rank(input), rank(input)).
+            Default: ``None`` , all dimensions are reduced.
+        keepdim (bool, optional): If ``True`` , keep these reduced dimensions and the length is 1.
+            If ``False`` , don't keep these dimensions. Default : ``False`` .
+
+    Returns:
+        Tensor, the dtype is bool.
+
+        - If `dim` is ``None`` , and `keepdim` is ``False`` ,
+          the output is a 0-D Tensor representing the "logical AND" of all elements in the input Tensor.
+        - If `dim` is int, such as 2, and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_1, input_3, ..., input_R)`.
+        - If `dim` is tuple(int), such as (2, 3), and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_1, input_4, ..., input_R)`.
+        - If `dim` is 1-D Tensor, such as [2, 3], and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_1, input_4, ..., input_R)`.
+
+    Raises:
+        TypeError: If `keepdim` is not a bool.
+        TypeError: If `input` is not a Tensor.
+        TypeError: If `dim` is not one of the following: int, tuple, list or Tensor.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import numpy as np
+        >>> from mindspore import Tensor, mint
+        >>> x = Tensor(np.array([[True, False], [True, True]]))
+        >>> # case 1: Reduces a dimension by the "logicalAND" of all elements in the dimension.
+        >>> output = mint.all(x, keepdim=True)
+        >>> print(output)
+        [[False]]
+        >>> print(output.shape)
+        (1, 1)
+        >>> # case 2: Reduces a dimension along axis 0.
+        >>> output = mint.all(x, dim=0)
+        >>> print(output)
+        [ True False]
+        >>> # case 3: Reduces a dimension along axis 1.
+        >>> output = mint.all(x, dim=1)
+        >>> print(output)
+        [False True]
+    """
+    return ops.function.math_func.all(input, dim, keepdim)
+
+
+def cat(tensors, dim=0):
+    r"""
+    Connect input tensors along with the given dimension.
+
+    The input data is a tuple or a list of tensors. These tensors have the same rank :math:`R`.
+    Set the given dimension as :math:`m`, and :math:`0 \le m < R`. Set the number of input tensors as :math:`N`.
+    For the :math:`i`-th tensor :math:`t_i`, it has the shape of :math:`(x_1, x_2, ..., x_{mi}, ..., x_R)`.
+    :math:`x_{mi}` is the :math:`m`-th dimension of the :math:`t_i`. Then, the shape of the output tensor is
+
+    .. math::
+
+        (x_1, x_2, ..., \sum_{i=1}^Nx_{mi}, ..., x_R)
+
+    Args:
+        tensors (Union[tuple, list]): A tuple or a list of input tensors.
+            Suppose there are two tensors in this tuple or list, namely t1 and t2.
+            To perform `concat` in the dimension 0 direction, except for the :math:`0`-th dimension,
+            all other dimensions should be equal, that is,
+            :math:`t1.shape[1] = t2.shape[1], t1.shape[2] = t2.shape[2], ..., t1.shape[R-1] = t2.shape[R-1]`,
+            where :math:`R` represents the rank of tensor.
+        dim (int): The specified dimension, whose value is in range :math:`[-R, R)`. Default: ``0`` .
+
+    Returns:
+        Tensor, the shape is :math:`(x_1, x_2, ..., \sum_{i=1}^Nx_{mi}, ..., x_R)`.
+        The data type is the same with `tensors`.
+
+    Raises:
+        TypeError: If `dim` is not an int.
+        ValueError: If `tensors` have different dimension of tensor.
+        ValueError: If `dim` not in range :math:`[-R, R)`.
+        ValueError: If tensor's shape in `tensors` except for `dim` are different.
+        ValueError: If `tensors` is an empty tuple or list.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import mindspore
+        >>> import numpy as np
+        >>> from mindspore import Tensor
+        >>> from mindspore import mint
+        >>> input_x1 = Tensor(np.array([[0, 1], [2, 1]]).astype(np.float32))
+        >>> input_x2 = Tensor(np.array([[0, 1], [2, 1]]).astype(np.float32))
+        >>> output = mint.cat((input_x1, input_x2))
+        >>> print(output)
+        [[0. 1.]
+         [2. 1.]
+         [0. 1.]
+         [2. 1.]]
+        >>> output = mint.cat((input_x1, input_x2), 1)
+        >>> print(output)
+        [[0. 1. 0. 1.]
+         [2. 1. 2. 1.]]
+    """
+    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.
+    And reduce a dimension of `input` along the specified `dim`. `keepdim`
+    determines whether the dimensions of the output and input are the same.
+
+    Note:
+        The `dim` with tensor type is only used for compatibility with older versions and is not recommended.
+
+    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.
+        dim (Union[int, tuple(int), list(int), Tensor]): The dimensions to reduce. Default: ``None`` ,
+            reduce all dimensions. Only constant value is allowed. Assume the rank of `input` is r,
+            and the value range is [-r,r).
+        keepdim (bool): If ``True`` , keep these reduced dimensions and the length is 1.
+            If ``False`` , don't keep these dimensions. Default: ``False`` .
+
+    Keyword Args:
+        dtype (:class:`mindspore.dtype`, optional): The desired data type of returned Tensor. Default: ``None`` .
+
+    Returns:
+        Tensor.
+
+        - If `dim` is ``None`` , and `keepdim` is ``False`` ,
+          the output is a 0-D tensor representing the product of all elements in the input tensor.
+        - If `dim` is int, set as 1, and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_0, input_2, ..., input_R)`.
+        - If `dim` is tuple(int) or list(int), set as (1, 2), and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_0, input_3, ..., input_R)`.
+        - If `dim` is 1-D Tensor, set as [1, 2], and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_0, input_3, ..., input_R)`.
+
+    Raises:
+        TypeError: If `input` is not a Tensor.
+        TypeError: If `dim` is not one of the following: int, tuple, list or Tensor.
+        TypeError: If `keepdim` is not a bool.
+        ValueError: If `dim` is out of range.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import mindspore
+        >>> import numpy as np
+        >>> from mindspore import Tensor, mint
+        >>> x = Tensor(np.random.randn(3, 4, 5, 6).astype(np.float32))
+        >>> output = mint.mean(x, 1, keepdim=True)
+        >>> result = output.shape
+        >>> print(result)
+        (3, 1, 5, 6)
+        >>> # case 1: Reduces a dimension by averaging all elements in the dimension.
+        >>> x = Tensor(np.array([[[2, 2, 2, 2, 2, 2], [2, 2, 2, 2, 2, 2], [2, 2, 2, 2, 2, 2]],
+        ... [[4, 4, 4, 4, 4, 4], [5, 5, 5, 5, 5, 5], [6, 6, 6, 6, 6, 6]],
+        ... [[6, 6, 6, 6, 6, 6], [8, 8, 8, 8, 8, 8], [10, 10, 10, 10, 10, 10]]]),
+        ... mindspore.float32)
+        >>> output = mint.mean(x)
+        >>> print(output)
+        5.0
+        >>> print(output.shape)
+        ()
+        >>> # case 2: Reduces a dimension along the axis 0
+        >>> output = mint.mean(x, 0, True)
+        >>> print(output)
+        [[[4. 4. 4. 4. 4. 4.]
+          [5. 5. 5. 5. 5. 5.]
+          [6. 6. 6. 6. 6. 6.]]]
+        >>> # case 3: Reduces a dimension along the axis 1
+        >>> output = mint.mean(x, 1, True)
+        >>> print(output)
+        [[[2. 2. 2. 2. 2. 2.]]
+         [[5. 5. 5. 5. 5. 5.]]
+         [[8. 8. 8. 8. 8. 8.]]]
+        >>> # case 4: Reduces a dimension along the axis 2
+        >>> output = mint.mean(x, 2, True)
+        >>> print(output)
+        [[[ 2.]
+          [ 2.]
+          [ 2.]]
+         [[ 4.]
+          [ 5.]
+          [ 6.]]
+         [[ 6.]
+          [ 8.]
+          [10.]]]
+    """
+    return ops.function.math_func.mean_ext(input, axis=dim, keep_dims=keepdim, dtype=dtype)
+
+
+def prod(input, dim=None, keepdim=False, *, dtype=None):
+    r"""
+    Reduces a dimension of a tensor by multiplying all elements in the dimension, by default. And also can
+    reduce a dimension of `input` along the `dim`. Determine whether the dimensions of the output and input are the
+    same by controlling `keepdim`.
+
+    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.
+        dim (int): The dimensions to reduce. Default: ``None`` , reduce all dimensions. Only constant value is allowed.
+            Assume the rank of `x` is r, and the value range is [-r,r).
+        keepdim (bool): If ``True`` , keep these reduced dimensions and the length is 1.
+            If ``False`` , don't keep these dimensions. Default: ``False`` .
+
+    Keyword Args:
+        dtype (:class:`mindspore.dtype`, optional): The desired data type of returned Tensor. Default: ``None`` .
+
+    Returns:
+        Tensor.
+
+        - If `dim` is ``None`` , and `keepdim` is ``False`` ,
+          the output is a 0-D tensor representing the product of all elements in the input tensor.
+        - If `dim` is int, set as 1, and `keepdim` is ``False`` ,
+          the shape of output is :math:`(input_0, input_2, ..., input_R)`.
+
+    Raises:
+        TypeError: If `input` is not a Tensor.
+        TypeError: If `dim` is not int.
+        TypeError: If `keepdim` is not a bool.
+        ValueError: If `dim` is out of range.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import mindspore
+        >>> import numpy as np
+        >>> from mindspore import Tensor, mint
+        >>> x = Tensor(np.random.randn(3, 4, 5, 6).astype(np.float32))
+        >>> output = mint.prod(x, 1, keepdim=True)
+        >>> result = output.shape
+        >>> print(result)
+        (3, 1, 5, 6)
+        >>> # case 1: Reduces a dimension by multiplying all elements in the dimension.
+        >>> x = Tensor(np.array([[[1, 1, 1, 1, 1, 1], [2, 2, 2, 2, 2, 2], [3, 3, 3, 3, 3, 3]],
+        ...                      [[4, 4, 4, 4, 4, 4], [5, 5, 5, 5, 5, 5], [6, 6, 6, 6, 6, 6]],
+        ...                      [[7, 7, 7, 7, 7, 7], [8, 8, 8, 8, 8, 8], [9, 9, 9, 9, 9, 9]]]), mindspore.float32)
+        >>> output = mint.prod(x)
+        >>> print(output)
+        2.2833798e+33
+        >>> print(output.shape)
+        ()
+        >>> # case 2: Reduces a dimension along axis 0.
+        >>> output = mint.prod(x, 0, True)
+        >>> print(output)
+        [[[ 28.  28.  28.  28.  28.  28.]
+          [ 80.  80.  80.  80.  80.  80.]
+          [162. 162. 162. 162. 162. 162.]]]
+        >>> # case 3: Reduces a dimension along axis 1.
+        >>> output = mint.prod(x, 1, True)
+        >>> print(output)
+        [[[  6.   6.   6.   6.   6.   6.]]
+         [[120. 120. 120. 120. 120. 120.]]
+         [[504. 504. 504. 504. 504. 504.]]]
+        >>> # case 4: Reduces a dimension along axis 2.
+        >>> output = mint.prod(x, 2, True)
+        >>> print(output)
+        [[[1.00000e+00]
+          [6.40000e+01]
+          [7.29000e+02]]
+         [[4.09600e+03]
+          [1.56250e+04]
+          [4.66560e+04]]
+         [[1.17649e+05]
+          [2.62144e+05]
+          [5.31441e+05]]]
+    """
+    return ops.auto_generate.prod_ext(input, axis=dim, keep_dims=keepdim, dtype=dtype)
+
+
+def ones(size, *, dtype=None):
+    r"""
+    Creates a tensor filled with value ones.
+
+    Creates a tensor with shape described by the first argument and fills it with value ones in type of the second
+    argument.
+
+    Args:
+        size (Union[tuple[int], list[int], int, Tensor]): The specified shape of output tensor. Only positive integer or
+            tuple or Tensor containing positive integers are allowed. If it is a Tensor,
+            it must be a 0-D or 1-D Tensor with int32 or int64 dtypes.
+
+    Keyword Args:
+        dtype (:class:`mindspore.dtype`, optional): The specified type of output tensor. If `dtype` is ``None`` ,
+            `mindspore.float32` will be used. Default: ``None`` .
+
+    Returns:
+        Tensor, whose dtype and size are defined by input.
+
+    Raises:
+        TypeError: If `size` is neither an int nor an tuple/list/Tensor of int.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import mindspore
+        >>> from mindspore import mint
+        >>> output = mint.ones((2, 2), dtype=mindspore.float32)
+        >>> print(output)
+        [[1. 1.]
+         [1. 1.]]
+    """
+    return ops.auto_generate.ones(size, dtype)
+
+
+def permute(input, dims):
+    """
+    Permutes the dimensions of the input tensor according to input `dims` .
+
+    Args:
+        input (Tensor): Input Tensor.
+        dims (tuple(int)): The order of the dimensions. Permute rearranges the `input` according
+            to the order of the `dims`.
+
+    Returns:
+        Tensor, has the same dimension as input tensor, with `axis` suitably permuted.
+
+    Raises:
+        ValueError: If `dims` is None.
+        ValueError: If the number of elements of `dims` is not equal to `input` ndim.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import mindspore
+        >>> import numpy as np
+        >>> from mindspore import Tensor, mint
+        >>> input_x = Tensor(np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]]), mindspore.float32)
+        >>> input_perm = (0, 2, 1)
+        >>> print(mint.permute(input_x, input_perm))
+        [[[ 1.  4.]
+          [ 2.  5.]
+          [ 3.  6.]]
+         [[ 7. 10.]
+          [ 8. 11.]
+          [ 9. 12.]]]
+    """
+    return ops.functional.permute(input, dims)
+
+
+def split(tensor, split_size_or_sections, dim=0):
+    """
+    Splits the Tensor into chunks along the given dim.
+
+    Args:
+        tensor (Tensor): A Tensor to be divided.
+        split_size_or_sections (Union[int, tuple(int), list(int)]):
+            If `split_size_or_sections` is an int type, `tensor` will be split into equally sized chunks,
+            each chunk with size `split_size_or_sections`. Last chunk will be smaller than `split_size_or_sections`
+            if `tensor.shape[dim]` is not divisible by `split_size_or_sections`.
+            If `split_size_or_sections` is a list type, then `tensor` will be split into len(split_size_or_sections)
+            chunks with sizes `split_size_or_sections` along the given `dim`.
+        dim (int): The dim along which to split. Default: ``0`` .
+
+    Returns:
+        A tuple of sub-tensors.
+
+    Raises:
+        TypeError: If argument `tensor` is not Tensor.
+        TypeError: If argument `dim` is not int.
+        ValueError: If argument `dim` is out of range of :[-tensor.ndim, tensor.ndim).
+        TypeError: If each element in `split_size_or_sections` is not integer.
+        TypeError: If argument `split_size_or_sections` is not int, tuple(int) or list(int).
+        ValueError: The sum of `split_size_or_sections` is not equal to x.shape[dim].
+
+    Supported Platforms:
+        ``Ascend``
+
+    Examples:
+        >>> import numpy as np
+        >>> from mindspore import ops, Tensor
+        >>> input_x = np.arange(9).astype("float32")
+        >>> output = ops.split(Tensor(input_x), 3)
+        >>> print(output)
+        (Tensor(shape=[3], dtype=Float32, value= [ 0.00000000e+00,  1.00000000e+00,  2.00000000e+00]),
+         Tensor(shape=[3], dtype=Float32, value= [ 3.00000000e+00,  4.00000000e+00,  5.00000000e+00]),
+         Tensor(shape=[3], dtype=Float32, value= [ 6.00000000e+00,  7.00000000e+00,  8.00000000e+00]))
+    """
+    return ops.function.array_func.split_ext(tensor, split_size_or_sections, dim)
+
+
+def sqrt(input):
+    r"""
+    Returns sqrt of a tensor element-wise.
+
+    .. math::
+
+        out_{i} = \sqrt{input_{i}}
+
+    Args:
+        input (Tensor): The input tensor with a dtype of number.Number.
+
+    Returns:
+        Tensor, has the same shape as the `input`.
+
+    Raises:
+        TypeError: If `input` is not a Tensor.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import mindspore
+        >>> import numpy as np
+        >>> from mindspore import Tensor, mint
+        >>> input = Tensor(np.array([1.0, 4.0, 9.0]), mindspore.float32)
+        >>> output = mint.sqrt(input)
+        >>> print(output)
+        [1. 2. 3.]
+    """
+    return ops.auto_generate.sqrt(input)
+
+
+def sub(input, other, *, alpha=1):
+    r"""
+    Subtracts scaled other value from input Tensor.
+
+    .. math::
+
+        out_{i} = input_{i} - alpha \times other_{i}
+
+    Note:
+        - When the two inputs have different shapes,
+          they must be able to broadcast to a common shape.
+        - The two inputs and alpha comply with the implicit type conversion rules to make the data types
+          consistent.
+
+    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/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/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.
+
+    Returns:
+        Tensor with a shape that is the same as the broadcasted shape of the input `input` and `other`,
+        and the data type is the one with higher precision or higher digits among the two inputs and alpha.
+
+    Raises:
+        TypeError: If the type of `input`, `other`, or `alpha` is not one of the following: Tensor, number.Number, bool.
+        TypeError: If `alpha` is of type float but `input` and `other` are not of type float.
+        TypeError: If `alpha` is of type bool but `input` and `other` are not of type bool.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import numpy as np
+        >>> import mindspore
+        >>> from mindspore import Tensor
+        >>> from mindspore import mint
+        >>> x = Tensor(np.array([4, 5, 6]).astype(np.float32))
+        >>> y = Tensor(1, mindspore.int32)
+        >>> alpha = 0.5
+        >>> output = mint.sub(x, y, alpha=alpha)
+        >>> print(output)
+        [3.5 4.5 5.5]
+        >>> # the data type of x is float32, the data type of y is int32,
+        >>> # alpha is a float, and the output is the data format of higher precision float32.
+        >>> print(output.dtype)
+        Float32
+    """
+    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`.
+
+    Args:
+        size (Union[tuple[int], list[int], int, Tensor]): The specified shape of output tensor. Only positive integer or
+            tuple or Tensor containing positive integers are allowed. If it is a Tensor,
+            it must be a 0-D or 1-D Tensor with int32 or int64 dtypes.
+
+    Keyword Args:
+        dtype (:class:`mindspore.dtype`, optional): The specified type of output tensor. If `dtype` is ``None`` ,
+            mindspore.float32 will be used. Default: ``None`` .
+
+    Returns:
+        Tensor, whose dtype and size are defined by input.
+
+    Raises:
+        TypeError: If `size` is neither an int nor an tuple/list/Tensor of int.
+
+    Supported Platforms:
+        ``Ascend`` ``GPU`` ``CPU``
+
+    Examples:
+        >>> import mindspore
+        >>> from mindspore import mint
+        >>> output = mint.zeros((2, 2), dtype=mindspore.float32)
+        >>> print(output)
+        [[0. 0.]
+         [0. 0.]]
+    """
+    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',
+    'isclose',
+    # 1
+    'div',
+    'divide',
+    'topk',
+    'roll',
+    # 2
+    'sin',
+    # 3
+    'clamp',
+    'xlogy',
+    # 4
+    'sinc',
+    'sinh',
+    'cosh',
+    # 5
+    'cumsum',
+    # 6
+    'stack',
+    # 7
+    'zeros',
+    # 8
+    'transpose',
+    'swapaxes',
+    # 9
+
+    # 10
+    'ne',
+    # 11
+    'unsqueeze',
+    # 12
+    "repeat_interleave",
+    # 13
+    "flip",
+    # 14
+
+    # 15
+    'flatten',
+    # 16
+    'matmul',
+    'bmm',
+    # 17
+    'mean',
+    # 18
+    'sum',
+    # 19
+    'log',
+    # 20
+    'prod',
+    # 21
+    'mul',
+    # 22
+
+    # 23
+
+    # 24
+
+    # 25
+    'greater',
+    'gt',
+    # 26
+    'eq',
+    # 27
+    'reciprocal',
+    # 28
+    'exp',
+    # 29
+    'sqrt',
+    # 30
+    'searchsorted',
+    # 31
+    'cummax',
+    'cummin',
+    'sub',
+    # 33
+    'split',
+    # 34
+
+    # 35
+    'erfinv',
+    # 36
+
+    # 37
+    'nonzero',
+    # 38
+
+    # 39
+
+    # 40
+    'any',
+    # 41
+    'add',
+    # 42
+    'argmax',
+    # 43
+    'cat',
+    # 44
+    'cos',
+    # 45
+    'concat',
+    # 46
+    'bitwise_and',
+    'bitwise_or',
+    'bitwise_xor',
+    # 47
+    'max',
+    # 48
+    'min',
+    # 49
+    'baddbmm',
+    # 50
+    'tile',
+    # 51
+    'permute',
+    # 52
+
+    # 53
+
+    # 54
+    'normal',
+    # 55
+    'cross',
+    # 56
+    'norm',
+    # 57
+    'broadcast_to',
+    # 58
+    'greater_equal',
+    # 59
+    'square',
+    # 60
+    'all',
+    # 61
+    'rsqrt',
+    # 62
+    'maximum',
+    # 63
+    'minimum',
+    # 64
+
+    # 65
+    'logical_and',
+    # 66
+    'logical_not',
+    # 67
+    'logical_or',
+    # 68
+    'logical_xor',
+    # 69
+    'less_equal',
+    'le',
+    # 70
+    'negative',
+    'neg',
+    # 71
+    'isfinite',
+    # 72
+
+    # 73
+    'ceil',
+    # 74
+    'sort',
+    # 75
+    'less',
+    'lt',
+    # 76
+    'pow',
+    # 77
+
+    # 78
+    'arange',
+
+    # 79
+
+    # 80
+
+    # 81
+    'index_select',
+    # 82
+
+    # 83
+    'narrow',
+    # 84
+
+    'masked_select',
+
+    # 86
+    'select',
+
+    # 87
+
+    # 88
+    'chunk',
+    # 89
+
+    # 90
+
+    # 91
+
+    # 92
+
+    # 93
+
+    # 94
+    'tanh',
+    # 95
+
+    # 96
+
+    # 97
+
+    # 98
+
+    # 99
+
+    # 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',
+]
+
+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__)