diff --git a/python_module/megengine/functional/math.py b/python_module/megengine/functional/math.py index c92f9890539abd912e5a699f1f163bad3e64e505..2ebac2f60c8192c3a343e7c9f55af9f12a6b5d6c 100644 --- a/python_module/megengine/functional/math.py +++ b/python_module/megengine/functional/math.py @@ -190,7 +190,6 @@ def sqrt(inp: Tensor) -> Tensor: Outputs: .. testoutput:: - :options: +NUMBER [[0. 1. 1.4142] [1.7321 2. 2.2361 ]] diff --git a/python_module/megengine/functional/nn.py b/python_module/megengine/functional/nn.py index a194d9c656eb9ead3ec9ad92c377a8366b4076de..d781fef1a14320d00f97d3a13a61daec866e60dc 100644 --- a/python_module/megengine/functional/nn.py +++ b/python_module/megengine/functional/nn.py @@ -636,7 +636,6 @@ def interpolate( Outputs: .. testoutput:: - :options: +NUMBER [[[[1. 1.25 1.75 2. ] [1.5 1.75 2.25 2.5 ] diff --git a/python_module/megengine/functional/sort.py b/python_module/megengine/functional/sort.py index f306b1fe865772a57bfc3a031863ae1452003815..70597d52899d5108d7cad69c155c7650493b0409 100644 --- a/python_module/megengine/functional/sort.py +++ b/python_module/megengine/functional/sort.py @@ -39,7 +39,6 @@ def argsort(inp: Tensor, descending: bool = False) -> Tuple[Tensor, Tensor]: Outputs: .. testoutput:: - :options: +NUMBER [1. 2.] [0 1] @@ -93,7 +92,6 @@ def top_k( Outputs: .. testoutput:: - :options: +NUMBER [1. 2. 3. 4. 5.] [7 0 6 1 5] diff --git a/python_module/megengine/functional/utils.py b/python_module/megengine/functional/utils.py index 5a54756d1b756f687dad6f082e5fc7d953262338..b490eb2f3b42951950f3feb9b6d1fe92b1d008c1 100644 --- a/python_module/megengine/functional/utils.py +++ b/python_module/megengine/functional/utils.py @@ -50,7 +50,6 @@ def accuracy(logits: Tensor, target: Tensor, topk: Union[int, Iterable[int]] = 1 Outputs: .. testoutput:: - :options: +NUMBER [0.] [0.375] """ diff --git a/python_module/megengine/module/activation.py b/python_module/megengine/module/activation.py index 73a7efb94115249a6de1295bb017c92b41653085..8240c488b91d64df02935d9cc5b3e86d57b1c3c3 100644 --- a/python_module/megengine/module/activation.py +++ b/python_module/megengine/module/activation.py @@ -20,7 +20,7 @@ class Softmax(Module): .. math:: \text{Softmax}(x_{i}) = \frac{exp(x_i)}{\sum_j exp(x_j)} - It is applied to an n-dimensional input Tensor and rescaling them so that the elements of the + It is applied to an n-dimensional input Tensor and rescaling them so that the elements of the n-dimensional output Tensor lie in the range of `[0, 1]` and sum to 1. :param axis: An axis along which softmax will be applied. By default, @@ -137,8 +137,8 @@ class PReLU(Module): ax, & \text{ otherwise } \end{cases} - Here :math:`a` is a learnable parameter. When called without arguments, `PReLU()` uses - a single paramter :math:`a` across all input channel. If called with `PReLU(num_of_channels)`, + Here :math:`a` is a learnable parameter. When called without arguments, `PReLU()` uses + a single paramter :math:`a` across all input channel. If called with `PReLU(num_of_channels)`, a seperate :math:`a` is used for each input channle. :param num_parameters: number of :math:`a` to learn, there is only two @@ -218,7 +218,6 @@ class LeakyReLU(Module): Outputs: .. testoutput:: - :options: +NUMBER [-0.08 -0.12 6. 10. ] diff --git a/python_module/megengine/module/embedding.py b/python_module/megengine/module/embedding.py index 806ae0e427880906717ef4719e91c9a6203bf609..976ac1250d8609002022c8f0be7621f033d53363 100644 --- a/python_module/megengine/module/embedding.py +++ b/python_module/megengine/module/embedding.py @@ -21,7 +21,7 @@ class Embedding(Module): A simple lookup table that stores embeddings of a fixed dictionary and size. This module is often used to store word embeddings and retrieve them using indices. - The input to the module is a list of indices, and the output is the corresponding word embeddings. + The input to the module is a list of indices, and the output is the corresponding word embeddings. The indices should less than num_embeddings. :param num_embeddings: size of embedding dictionary. @@ -138,7 +138,6 @@ class Embedding(Module): Outputs: .. testoutput:: - :options: +NUMBER [[[1.2 2.3 3.4 4.5 5.6] [0.1 1.1 2.1 3.1 4.1]