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

mindspore/mint/special/__init__.py

@@ -0,0 +1,63 @@
+# 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
+
+from mindspore.ops.auto_generate import erfc, expm1, sinc, log1p, round_op
+from mindspore.ops.function.nn_func import log_softmax_ext as log_softmax
+
+
+def round(input):
+    r"""
+    Returns half to even of a tensor element-wise.
+
+    .. math::
+        out_i \approx input_i
+
+    .. note::
+        The input data types supported by the Ascend platform include
+        bfloat16 (Atlas training series products are not supported), float16, float32, float64, int32, and int64.
+
+    Args:
+        input (Tensor): The input tensor.
+
+    Returns:
+        Tensor, has the same shape and type 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, ops
+        >>> input = Tensor(np.array([0.8, 1.5, 2.3, 2.5, -4.5]), mindspore.float32)
+        >>> output = mint.special.round(input)
+        >>> print(output)
+        [ 1.  2.  2.  2. -4.]
+    """
+    return round_op(input, 0)
+
+__all__ = [
+    'erfc',
+    'expm1',
+    'round',
+    'sinc',
+    'log1p',
+    'log_softmax',
+]

mindspore/multiprocessing/__init__.py

@@ -20,6 +20,7 @@ import types
 import signal
 import multiprocessing as mp
 from multiprocessing import *
+from multiprocessing import pool as mp_pool
 from mindspore._c_expression import fork_utils
 
 __all__ = []
@@ -59,7 +60,7 @@ class Process(mp.Process): # pylint: disable=function-redefined
 _MsProcess = Process
 
 
-class Pool(mp.pool.Pool): # pylint: disable=function-redefined, abstract-method
+class Pool(mp_pool.Pool):  # pylint: disable=function-redefined, abstract-method
     """
     Trigger fork callbacks by overriding native multiprocessing methods.
     """

mindspore/nn/__init__.py

@@ -31,7 +31,6 @@ from mindspore.nn.wrap import *
 from mindspore.nn.grad import Jvp, Vjp
 from mindspore.nn.sparse import *
 from mindspore.nn.reinforcement import *
-from mindspore.nn import extend
 
 __all__ = ["Cell", "GraphCell"]
 __all__.extend(layer.__all__)