mindspore 2.5.0__cp310-cp310-win_amd64.whl → 2.6.0__cp310-cp310-win_amd64.whl

mindspore/ops/operations/inner_ops.py

@@ -87,7 +87,7 @@ class NoRepeatNGram(Primitive):
     Because 3 consecutive words [2, 3, 2] do not appear twice in the word sequence.
 
     Args:
-        ngram_size (int): Size of n-grams, must be greater than 0. Default: ``1`` .
+        ngram_size (int, optional): Size of n-grams, must be greater than 0. Default: ``1`` .
 
     Inputs:
         - **state_seq** (Tensor) - n-gram word series, a 3-D tensor with shape: :math:`(batch\_size, beam\_width, m)`.

mindspore/ops/operations/linalg_ops.py

@@ -19,64 +19,7 @@ from __future__ import absolute_import
 from mindspore import _checkparam as Validator
 from mindspore.ops.primitive import Primitive
 from mindspore.ops.primitive import prim_attr_register
-from ..auto_generate import Geqrf
-
-
-class Svd(Primitive):
-    """
-    Computes the singular value decompositions of one or more matrices.
-
-    Refer to :func:`mindspore.ops.svd` for more details.
-
-    Args:
-        full_matrices (bool, optional): If ``True`` , compute full-sized :math:`U` and :math:`V`. If ``False``,
-                                        compute only the leading P singular vectors, with P is the minimum of M and N.
-                                        Default: ``False`` .
-        compute_uv (bool, optional): If ``True`` , compute the left and right singular vectors.
-                                     If ``False`` , compute only the singular values. Default: ``True`` .
-
-    Inputs:
-        - **input** (Tensor) - Tensor of the matrices to be decomposed. The shape should be :math:`(*, M, N)`,
-          the supported dtype are float32 and float64.
-
-    Outputs:
-        - **s**  (Tensor) - Singular values. The shape is :math:`(*, P)`.
-        - **u**  (Tensor) - Left singular vectors. If `compute_uv` is ``False`` , u will be zero value.
-          The shape is :math:`(*, M, P)`. If `full_matrices` is ``True`` , the shape will be :math:`(*, M, M)`.
-        - **v**  (Tensor) - Right singular vectors. If `compute_uv` is ``False`` , v will be zero value.
-          The shape is :math:`(*, N, P)`. If `full_matrices` is ``True`` , the shape will be :math:`(*, N, N)`.
-
-    Supported Platforms:
-        ``GPU`` ``CPU``
-
-    Examples:
-        >>> import numpy as np
-        >>> import mindspore as ms
-        >>> from mindspore import Tensor
-        >>> from mindspore import ops
-        >>> ms.set_device(device_target="CPU")
-        >>> svd = ops.Svd(full_matrices=True, compute_uv=True)
-        >>> a = Tensor(np.array([[1, 2], [-4, -5], [2, 1]]).astype(np.float32))
-        >>> s, u, v = svd(a)
-        >>> print(s)
-        [7.0652843 1.040081 ]
-        >>> print(u)
-        [[ 0.30821905 -0.48819482 0.81649697]
-         [-0.90613353  0.11070572 0.40824813]
-         [ 0.2896955   0.8656849  0.4082479 ]]
-        >>> print(v)
-        [[ 0.63863593 0.769509  ]
-         [ 0.769509  -0.63863593]]
-    """
-
-    @prim_attr_register
-    def __init__(self, full_matrices=False, compute_uv=True):
-        super().__init__(name="Svd")
-        self.init_prim_io_names(inputs=['a'], outputs=['s', 'u', 'v'])
-        self.full_matrices = Validator.check_value_type("full_matrices", full_matrices, [bool], self.name)
-        self.compute_uv = Validator.check_value_type("compute_uv", compute_uv, [bool], self.name)
-        self.add_prim_attr('full_matrices', self.full_matrices)
-        self.add_prim_attr('compute_uv', self.compute_uv)
+from ..auto_generate import Geqrf, Svd
 
 
 class Eigh(Primitive):

mindspore/ops/operations/manually_defined/_inner.py

@@ -66,7 +66,7 @@ class TensorReport(Primitive):
     @prim_attr_register
     def __init__(self):
         """Initialize TensorReport"""
-        self.add_prim_attr("side_effect_io", True)
+        self.add_prim_attr("side_effect_mem", True)
         self.add_prim_attr("channel_name", "ms_tensor_report")
 
     def __call__(self, file, input_x):