27 kB

Raw Permalink Blame History

MindSpore API注释规范

MindSpore API注释规范

MindSpore API注释规范

总体说明

MindSpore Python代码注释遵循Google Python Style Guide，由Sphinx工具自动生成API文档，注释样例和支持情况可参考Example Google Style Python Docstrings和Support for NumPy and Google style docstrings。
MindSpore C++代码需按照命名空间的设计，分别编写Markdown文件。

Python API注释规范

注释格式

类和方法的注释都采用如下格式：

Summary.

More elaborate description.

.. warning::
    Warning description.

Note:
    Note description.

Args:
    Arg1 (Type): Description. Default: xxx.
    Arg2 (Type): Description.

        - Sub-argument1 or Value1 of Arg2: Description.
        - Sub-argument2 or Value2 of Arg2: Description.

Returns:
    Type, description.

Raises:
    Type: Description.

Examples:
    >>> Sample Code

其中，

Summary：简单描述该接口的功能。若该描述为动词开头，模块内需统一用第一人称（即动词原形）或第三人称（即动词后加s），推荐用第一人称。
More elaborate description：详细描述该接口的功能和如何使用等信息。
warning：描述使用该接口时需要警告的事项，以免造成负面影响。
Note：描述使用该接口时需要注意的事项。特别注意不能写成Notes。
Args：接口参数信息，包含参数名、参数类型、取值范围、默认值等。
Returns：返回值信息，包含返回值类型等。
Raises：异常信息，包含异常类型、含义等。
Examples：样例代码。

针对算子和Cell的注释，需要在Examples前添加Inputs、Outputs和Supported Platforms三项内容。

Inputs和Outputs：用于描述实例化后，算子的输入和输出的类型和shape，输入名可以和样例相同。建议在注释中给出对应的数学公式。
Supported Platforms：用于描述算子支持的硬件平台，名称前后需添加``，存在多个时使用空格隔开。

Inputs:
    - **input_name1** (Type) - Description.
    - **input_name2** (Type) - Description.

Outputs:
    Type and shape, description.

Supported Platforms:
    ``Ascend`` ``GPU`` ``CPU``

注意事项

整体要求

类或方法必须书写的注释项有：Summary、Args、Returns和Raises。如果函数中没有相关信息（如Args、Returns和Raises等），不需要写None（如Raises：None），直接省略注释项即可。
以目录为粒度生成API时，__init__文件头部的注释内容会呈现在网页开头；以文件为粒度生成API时，该文件头部的注释内容会呈现在网页开头。这些注释内容需包含相应模块的整体说明。
注释中包含反斜杠时，需要将头部的"""改成r"""。
注释内容若为动词开头，模块内需统一用第一人称（即动词原形）或第三人称（即动词后加s），推荐用第一人称。
冒号要求：关键字（如Args、Returns等）后面有冒号":"；参数名（如Arg1、Arg2等）后面有冒号":"，冒号后需有空格。Summary和Returns的内容中不能包含冒号。

空行要求：

不同类型（如Args、Returns等）的内容之间需有空行，同种类型（如Arg1、Arg2等）的内容之间不需要空行。

High-Level API for Training or Testing.

Args:
    network (Cell): A training or testing network.
    loss_fn (Cell): Objective function, if `loss_fn` is None, the
        network should contain the logic of loss and grads calculation, and the logic
        of parallel if needed. Default: None.

Returns:
    function, original function.

采用无序或有序列表描述内容时，整个列表内容与上方内容之间需增加一个空行。

Args:
    amp_level (str): Option for argument `level` in `mindspore.amp.build_train_network`, level for mixed
        precision training. Supports ["O0", "O2", "O3", "auto"]. Default: "O0".

        - O0: Do not change.
        - O2: Cast network to float16, keep batchnorm run in float32, using dynamic loss scale.
        - O3: Cast network to float16, with additional property 'keep_batchnorm_fp32=False'.
        - auto: Set to level to recommended level in different devices. Set level to "O2" on GPU, set
          level to "O3" Ascend. The recommended level is choose by the export experience, cannot
          always generalize. User should specify the level for special network.

        "O2" is recommended on GPU, "O3" is recommended on Ascend.

空格要求：

Args和Raises内容的换行需要缩进4个空格。

Args:
    lr_power (float): Learning rate power controls how the learning rate decreases during training,
        must be less than or equal to zero. Use fixed learning rate if `lr_power` is zero.
    use_locking (bool): If True, the var and accumulation tensors will be protected from being updated.
        Default: False.

Raises:
    TypeError: If `lr`, `l1`, `l2`, `lr_power` or `use_locking` is not a float.
        If `use_locking` is not a bool.
        If dtype of `var`, `accum`, `linear` or `grad` is neither float16 nor float32.
        If dtype of `indices` is not int32.

以下内容的换行不需要缩进，与上一行的正文起始位置对齐。

Args的子参数或取值

Args:
    parallel_mode (str): There are five kinds of parallel modes, "stand_alone", "data_parallel",
        "hybrid_parallel", "semi_auto_parallel" and "auto_parallel". Default: "stand_alone".

        - stand_alone: Only one processor is working.
        - data_parallel: Distributes the data across different processors.
        - hybrid_parallel: Achieves data parallelism and model parallelism
          manually.
        - semi_auto_parallel: Achieves data parallelism and model parallelism by
          setting parallel strategies.

Inputs、Outputs和Returns等无序或有序列表内容

Inputs:
    - **var** (Parameter) - The variable to be updated. The data type must be float16 or float32.
    - **accum** (Parameter) - The accumulation to be updated, must be same data type and shape as `var`.
    - **linear** (Parameter) - the linear coefficient to be updated, must be same data type and shape
      as `var`.
    - **grad** (Tensor) - A tensor of the same type as `var`, for the gradient.
    - **indices** (Tensor) - A vector of indices in the first dimension of `var` and `accum`.
      The shape of `indices` must be the same as `grad` in the first dimension. The type must be int32.

Outputs:
    Tuple of 3 Tensor, the updated parameters.

    - **var** (Tensor) - Tensor, has the same shape
      and data type as `var`.
    - **accum** (Tensor) - Tensor, has the same shape
      and data type as `accum`.
    - **linear** (Tensor) - Tensor, has the same shape
      and data type as `linear`.

Note和warning

.. warning::
    This is warning text. Use a warning for information the user must
    understand to avoid negative consequences.

    If warning text runs over a line, make sure the lines wrap and are indented to
    the same level as the warning tag.

Args中参数名和类型的(之间需要有空格。

Args:
    lr (float): The learning rate value, must be positive.

Args注释说明

常见参数类型有：

基本数据类型：int、float、bool、str、list、dict、set、tuple、numpy.ndarray。

Args:
    arg1 (int): Some description.
    arg2 (float): Some description.
    arg3 (bool): Some description.
    arg4 (str): Some description.
    arg5 (list): Some description.
    arg6 (dict): Some description.
    arg7 (set): Some description.
    arg8 (tuple): Some description.
    arg9 (numpy.ndarray): Some description.

dtype：如果是mindspore.dtype里的值，写成mindspore.dtype，如果是numpy类型，写成numpy.dtype。其他按实际情况写。
```
Args:
    arg1 (mindspore.dtype): Some description.
```
一个参数有多个可选类型：Union[类型1, 类型2]，如Union[Tensor, Number]。
```
Args:
    arg1 (Union[Tensor, Number]): Some description.
```

list类型：list[具体类型]，如list[str]。

Args:
    arg1 (list[str]): Some description.

可选类型统一格式：(类型, optional)。

Args:
    arg1 (bool, optional): Some description.

其他类型：Tensor或其他具体类型或方法名。
```
Args:
    arg1 (Tensor): Some description.
```

Returns注释说明
- 如果返回值类型或维度发生变化，需要说明返回值与输入的关系。
- 多个返回值时，分行写，网页显示时不分行，无序列表的方式可支持分行。
```
Returns:
    - DatasetNode, the root node of the IR tree.
    - Dataset, the root dataset of the IR tree.
```

Examples注释说明

Examples中的内容需在每行代码开头加上>>>，多行代码（含类或函数定义、人为换行等）或空行的开头需加上...，输出结果行开头不需要加任何符号。

Examples:
    >>> import mindspore as ms
    >>> import mindspore.nn as nn
    >>> class Net(nn.Cell):
    ...     def __init__(self, dense_shape):
    ...         super(Net, self).__init__()
    ...         self.dense_shape = dense_shape
    ...     def construct(self, indices, values):
    ...         x = SparseTensor(indices, values, self.dense_shape)
    ...         return x.values, x.indices, x.dense_shape
    ...
    >>> indices = Tensor([[0, 1], [1, 2]])
    >>> values = Tensor([1, 2], dtype=ms.float32)
    >>> out = Net((3, 4))(indices, values)
    >>> print(out[0])
    [1. 2.]
    >>> print(out[1])
    [[0 1]
     [1 2]]
    >>> print(out[2])
    (3, 4)

Examples中需提供实际的代码，如果需提示参考其他Examples，请使用Note。
ops算子注释采用PyNative模式写作，可运行的需给出运行结果。
业界共识的情况可省略import，如np、nn等。
导入路径较长和必须自定义别名的导入需要加from xxx import xxx as something或import xxx，导入路径短的尽量放到代码中。

Inputs和Outputs注释说明
- 类型是Tensor时，需描述shape，并按:math:`(N, C, X)`格式写作。
公式
- 行公式（单独占一行，居中）
```
.. math::

   formula
```
- 行内嵌公式（与其他同行文字显示在一起，不居中）
```
xxx :math:`formula` xxx
```
- 公式中带有含下划线的变量，且下划线后存在多个字母（如xxx_yyy），请根据实际需要选择以下其中一种方式。
  1. 多个字母用{}括起来（如xxx_{yyy}），可将下划线后的内容作为下标，显示为$xxx_{yyy}$。
  2. 在下划线前增加反斜杠（如xxx\_yyy），可将完整显示变量名称，显示为xxx_yyy。
父类方法的显示
- 默认不显示父类方法。
- 可通过在Sphinx工程rst文件的模块下添加:inherited-members:，指定需要显示父类方法，详细可参考https://www.sphinx-doc.org/en/master/usage/extensions/autodoc.html。
链接
- 只显示标题（如例子中的name），不显示详细地址。
  
  引用的地方需这样写：
```
`name`_
```
  提供链接的地方需这样写：
```
.. _`name`: https://xxx
```
  注意：
  - 如有换行请注意缩进，参考下方具体示例。
  - https前需有空格。
  或者可以采用以下简化写法，只在引用的地方写即可。
```
`name <https://xxx>`_
```
- 直接显示详细地址。
```
https://xxx
```

表格（详细可参考https://sublime-and-sphinx-guide.readthedocs.io/en/latest/tables.html#list-table-directive）

.. list-table:: Title            # 表格标题
:widths: 25 25 25               # 表格列宽
:header-rows: 1

* - Heading row 1, column 1     # 表头
    - Heading row 1, column 2
    - Heading row 1, column 3
* - Row 1, column 1
    -                             # 表格内容为空
    - Row 1, column 3
* - Row 2, column 1
    - Row 2, column 2
    - Row 2,
                                # 表格内容如需换行，在中间增加一个空行
    column 3

显示效果：

详细说明默认不换行，如需换行，需以列表或code-block的方式写作。
- 列表方式：
```
- Content1
- Content2
- Content3
```
- code-block方式：
```
.. code-block::

Content1
Content2
Content3
```
在注释中引用其他接口。
- 引用class。
  
  只写接口名：
```
:class:`AdamNoUpdateParam`
```
  若存在重复接口名，则需引用完整模块名和类名：
```
:class:`mindspore.ops.LARS`
```
- 引用function，必须写上完整模块名和函数名。
```
:func:`mindspore.compression.quant.create_quant_config`
```

接口描述中，变量名或接口名使用符号`包裹，变量值使用符号'或"包裹。

变量名或接口名。

This part is a more detailed overview of `Mul` operation. For more details about Quantization,
please refer to the implementation of subclass of `Observer`.

Other losses derived from this should implement their own `construct` and use method `self.get_loss`
to apply reduction to loss values.

变量值。

If `reduction` is not one of 'none', 'mean', 'sum'.

废弃算子需要写明建议使用的接口，以及需要在支持平台中写上Deprecated。

class BasicLSTMCell(PrimitiveWithInfer):
    """
    It's similar to operator :class:`DynamicRNN`. BasicLSTMCell will be deprecated in the future.
    Please use :class:`DynamicRNN` instead.

    Supported Platforms:
        Deprecated
    """

添加图片。

格式：.. image:: {name.png}。

其中{name.png}为图片名称，并将图片提交到https://gitee.com/mindspore/mindspore/tree/master/docs/api/api_python中对应模块的目录下。

如在mindspore.dataset.audio.transforms.FrequencyMasking接口注释中添加图片frequency_masking.png：
```
class FrequencyMasking(AudioTensorOperation):
"""
Some description.

.. image:: frequency_masking.png
"""
```
并将图片提交至：https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/dataset_audio/frequency_masking.png。

Python示例

类

class Tensor(Tensor_):
    """
    Tensor is used for data storage.

    Tensor inherits tensor object in C++.
    Some functions are implemented in C++ and some functions are implemented in Python.

    Args:
        input_data (Tensor, float, int, bool, tuple, list, numpy.ndarray): Input data of the tensor.
        dtype (:class:`mindspore.dtype`): Input data should be None, bool or numeric type defined in `mindspore.dtype`.
            The argument is used to define the data type of the output tensor. If it is None, the data type of the
            output tensor will be as same as the `input_data`. Default: None.

    Outputs:
        Tensor, with the same shape as `input_data`.

    Examples:
        >>> # initialize a tensor with input data
        >>> t1 = Tensor(np.zeros([1, 2, 3]), mindspore.float32)
        >>> assert isinstance(t1, Tensor)
        >>> assert t1.shape == (1, 2, 3)
        >>> assert t1.dtype == mindspore.float32
        ...
        >>> # initialize a tensor with a float scalar
        >>> t2 = Tensor(0.1)
        >>> assert isinstance(t2, Tensor)
        >>> assert t2.dtype == mindspore.float64
    """

    def __init__(self, input_data, dtype=None):
        ...

显示效果可访问这里。

方法

def ms_function(fn=None, obj=None, input_signature=None):
    """
    Create a callable MindSpore graph from a python function.

    This allows the MindSpore runtime to apply optimizations based on graph.

    Args:
        fn (Function): The Python function that will be run as a graph. Default: None.
        obj (Object): The Python Object that provides the information for identifying the compiled function. Default:
            None.
        input_signature (MetaTensor): The MetaTensor which describes the input arguments. The MetaTensor specifies
            the shape and dtype of the Tensor and they will be supplied to this function. If `input_signature`
            is specified, each input to `fn` must be a `Tensor`. And the input parameters of `fn` cannot accept
            `**kwargs`. The shape and dtype of actual inputs should keep the same as `input_signature`. Otherwise,
            TypeError will be raised. Default: None.

    Returns:
        Function, if `fn` is not None, returns a callable function that will execute the compiled function; If `fn` is
        None, returns a decorator and when this decorator invokes with a single `fn` argument, the callable function is
        equal to the case when `fn` is not None.

    Examples:
        >>> def tensor_add(x, y):
        ...     z = F.tensor_add(x, y)
        ...     return z
        ...
        >>> @ms_function
        ... def tensor_add_with_dec(x, y):
        ...     z = F.tensor_add(x, y)
        ...     return z
        ...
        >>> @ms_function(input_signature=(MetaTensor(mindspore.float32, (1, 1, 3, 3)),
        ...                               MetaTensor(mindspore.float32, (1, 1, 3, 3))))
        ... def tensor_add_with_sig(x, y):
        ...     z = F.tensor_add(x, y)
        ...     return z
        ...
        >>> x = Tensor(np.ones([1, 1, 3, 3]).astype(np.float32))
        >>> y = Tensor(np.ones([1, 1, 3, 3]).astype(np.float32))
        ...
        >>> tensor_add_graph = ms_function(fn=tensor_add)
        >>> out = tensor_add_graph(x, y)
        >>> out = tensor_add_with_dec(x, y)
        >>> out = tensor_add_with_sig(x, y)
    """
    ...

显示效果可访问这里。

公式

class Conv2d(_Conv):
    r"""
    2D convolution layer.

    Apply a 2D convolution over an input tensor which is typically of shape :math:`(N, C_{in}, H_{in}, W_{in})`,
    where :math:`N` is batch size, :math:`C_{in}` is channel number, and :math:`H_{in}, W_{in})` are height and width.
    For each batch of shape :math:`(C_{in}, H_{in}, W_{in})`, the formula is defined as:

    .. math::

        out_j = \sum_{i=0}^{C_{in} - 1} ccor(W_{ij}, X_i) + b_j,

    ...
    """

显示效果可访问这里。

链接

class BatchNorm(PrimitiveWithInfer):
    r"""
    Batch Normalization for input data and updated parameters.

    Batch Normalization is widely used in convolutional neural networks. This operation
    applies Batch Normalization over input to avoid internal covariate shift as described
    in the paper `Batch Normalization: Accelerating Deep Network Training by Reducing Internal
    Covariate Shift <https://arxiv.org/abs/1502.03167>`_. It rescales and recenters the
    features using a mini-batch of data and the learned parameters which can be described
    in the following formula,

    ...
    """

显示效果可访问这里。

C++ API注释规范

注释格式

所有接口注释都采用如下格式：

/// \brief Short description
///
/// Detailed description.
///
/// \note
/// Describe what to be aware of when using this interface.
///
/// \f[
/// math formula
/// \f]
/// XXX \f$ formulas in the line \f$ XXX
///
/// \param[in] Parameter_name meaning, range of values, other instructions.
///
/// \return Returns a description of the value, the cause of the error,
///     and the corresponding solution.
///
/// \par Example
/// \code
/// Example code
/// \endcode

其中：

\brief：简要描述。

/// \brief Function to create a CocoDataset.

Detailed description：详细描述。

///  Base class for all recognizable patterns.
///  We implement an Expression Template approach using static polymorphism based on
///  the Curiously Recurring Template Pattern (CRTP) which "achieves a similar effect
///  to the use of virtual functions without the costs..." as described in:
///  https://en.wikipedia.org/wiki/Expression_templates and
///  https://en.wikipedia.org/wiki/Curiously_recurring_template_pattern
///  The TryCapture function tries to capture the pattern with the given node.
///  The GetNode function builds a new node using the captured values.

\note：使用该接口的注意事项。

/// \note
/// The generated dataset has multi-columns

公式写法。

多行公式写法：

/// \f[
/// x>=y
/// \f]

行内公式写法，公式位于两个\f$之间：

/// \brief Computes the boolean value of \f$x>=y\f$ element-wise.

\param[in]：传入参数描述。

/// \param[in] weight Defines the width of memory to request
/// \param[in] height Defines the height of memory to request
/// \param[in] type Defines the data type of memory to request

\return：返回值描述。

/// \return Reference count of a certain memory currently.

示例代码，格式如下，\par Example作为前缀，示例代码位于\code和\endcode之间：

/// \par Example
/// \code
/// /* Set number of workers(threads) to process the dataset in parallel */
/// std::shared_ptr<Dataset> ds = ImageFolder(folder_path, true);
/// ds = ds->SetNumWorkers(16);
/// \endcode

注意事项

需要生成文档的接口注释内容统一使用///引导而不是使用//引导；
不要间断注释，空行使用///；
引用在C++ API中具有同名的外部名称时，避免生成错误链接，需要在前面添加@ref标识：
```
/// \brief Referring to @ref mindspore.nn.Cell for detail.
```

完整示例

/// \brief Function to create a MnistDataset.
/// \note The generated dataset has two columns ["image", "label"].
/// \param[in] dataset_dir Path to the root directory that contains the dataset.
/// \param[in] usage Part of dataset of MNIST, can be "train", "test" or "all" (default = "all").
/// \param[in] sampler Shared pointer to a sampler object used to choose samples from the dataset. If sampler is not
///     given, a `RandomSampler` will be used to randomly iterate the entire dataset (default = RandomSampler()).
/// \param[in] cache Tensor cache to use (default=nullptr which means no cache is used).
/// \return Shared pointer to the MnistDataset.
/// \par Example
/// \code
///      /* Define dataset path and MindData object */
///      std::string folder_path = "/path/to/mnist_dataset_directory";
///      std::shared_ptr<Dataset> ds = Mnist(folder_path, "all", std::make_shared<RandomSampler>(false, 20));
///
///      /* Create iterator to read dataset */
///      std::shared_ptr<Iterator> iter = ds->CreateIterator();
///      std::unordered_map<std::string, mindspore::MSTensor> row;
///      iter->GetNextRow(&row);
///
///      /* Note: In MNIST dataset, each dictionary has keys "image" and "label" */
///      auto image = row["image"];
/// \endcode
inline std::shared_ptr<MnistDataset> MS_API
Mnist(const std::string &dataset_dir, const std::string &usage = "all",
      const std::shared_ptr<Sampler> &sampler = std::make_shared<RandomSampler>(),
      const std::shared_ptr<DatasetCache> &cache = nullptr) {
  return std::make_shared<MnistDataset>(StringToChar(dataset_dir), StringToChar(usage), sampler, cache);
}

根据以上注释内容输出的API文档页面为Function mindspore::dataset::Coco。

27 kB Raw Permalink Blame History

MindSpore API注释规范

总体说明

Python API注释规范

注释格式

注意事项

Python示例

类

方法

公式

链接

C++ API注释规范

注释格式

注意事项

完整示例

27 kB

Raw Permalink Blame History