modify the format of files 1213

This commit is contained in:
huanxiaoling 2022-12-13 14:00:11 +08:00
parent 5b6dedf93b
commit 12ef0dcea9
7 changed files with 9 additions and 26 deletions

View File

@ -1,7 +1,7 @@
mindspore.JitConfig
====================
.. py:class:: mindspore.JitConfig(jit_level="O1", exc_mode=auto, **kwargs)
.. py:class:: mindspore.JitConfig(jit_level="O1", exc_mode="auto", **kwargs)
编译时所使用的JitConfig配置项。

View File

@ -1,7 +1,7 @@
mindspore.ops.addr
==================
.. py:function:: mindspore.ops.addr(vec1, vec2, beta=1, alpha=1)
.. py:function:: mindspore.ops.addr(x, vec1, vec2, beta=1, alpha=1)
计算 `vec1``vec2` 的外积,并将其添加到 `x` 中。
@ -14,6 +14,7 @@ mindspore.ops.addr
output = β x + α (vec1 ⊗ vec2)
参数:
- **x** (Tensor) - 需要相加的向量。Tensor的shape是 :math:`(N, M)`
- **vec1** (Tensor) - 第一个需要相乘的Tensorshape大小为 :math:`(N,)`
- **vec2** (Tensor) - 第二个需要相乘的Tensorshape大小为 :math:`(M,)`
- **beta** (scalar[int, float, bool], 可选) - `x` (β)的乘法器。 `beta` 必须是int或float或bool类型默认值1。

View File

@ -12,4 +12,4 @@ mindspore.ops.t
Tensor`x` 的转置。
异常:
- **TypeError** - `x` 的维度大于2。
- **ValueError** - `x` 的维度大于2。

View File

@ -7,7 +7,7 @@ mindspore.ops.zeros
参数:
- **shape** (Union[tuple[int], int]) - 用来描述所创建的Tensor的 `shape`
- **dtype** (:class:`mindspore.dtype`) - 用来描述所创建的Tensor的 `dtype`。如果为None那么将会使用mindspore.float32。默认值None。
- **dtype** (:class:`mindspore.dtype`, 可选) - 用来描述所创建的Tensor的 `dtype`。如果为None那么将会使用mindspore.float32。默认值None。
返回:
Tensordtype和shape由入参决定。

View File

@ -9314,7 +9314,7 @@ def sum(x, dim=None, keepdim=False, *, dtype=None):
Returns:
A Tensor, sum of elements over a given dim in `x`.
Raise:
Raises:
TypeError: If `x` is not a Tensor.
TypeError: If `dim` is not an int.
ValueError: If `dim` is not in the range :math:`[-x.ndim, x.ndim)` .

View File

@ -1825,24 +1825,6 @@ class Argmax(Primitive):
Refer to :func:`mindspore.ops.argmax` for more details.
If the shape of input tensor is :math:`(x_1, ..., x_N)`, the shape of the output tensor will be
:math:`(x_1, ..., x_{axis-1}, x_{axis+1}, ..., x_N)`.
Args:
axis (int): Axis where the Argmax operation applies to. Default: -1.
output_type (:class:`mindspore.dtype`): An optional data type of `mindspore.dtype.int32` and
`mindspore.dtype.int64`. Default: `mindspore.dtype.int32`.
Inputs:
- **input_x** (Tensor) - Input tensor. :math:`(N,*)` where :math:`*` means, any number of additional dimensions.
Outputs:
Tensor, whose dtype is determined by `output_type`.
Raises:
TypeError: If `axis` is not an int.
TypeError: If `output_type` is neither int32 nor int64.
Supported Platforms:
``Ascend`` ``GPU`` ``CPU``
@ -5751,8 +5733,8 @@ class Sort(Primitive):
- **x** (Tensor) - The input tensor of any dimension, with a type of float16 or float32.
Outputs:
y1(Tensor), a tensor whose values are the sorted values, with the same shape and data type as input.
y2(Tensor), the indices of the elements in the original input tensor. Data type is int32.
- **y1** (Tensor) - A tensor whose values are the sorted values, with the same shape and data type as input.
- **y2** (Tensor) - the indices of the elements in the original input tensor. Data type is int32.
Raises:
TypeError: If `axis` is not an int.