xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix LinSpace doc

This commit is contained in:
jiangjinsheng 2020-07-07 15:27:04 +08:00
parent 7304f02410
commit 0d22e64fa8
2 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mindspore/nn/layer/math.py
View File

@ -132,23 +132,19 @@ class Range(Cell):
class LinSpace(Cell):
r"""
Generates values in an interval. And return the corresponding interpolation accroding to assist.
Generates values in an interval.
Args:
- **start** (Union[int, float]) - The start of interval, With shape of 0-D.
- **stop** (Union[int, float]) - The end of interval, With shape of 0-D.
- **num** (int) - ticks number in the interval, the ticks include start and stop value.
With shape of 0-D.
start (Union[int, float]): The start of interval. With shape of 0-D.
stop (Union[int, float]): The end of interval. With shape of 0-D.
num (int): ticks number in the interval, the ticks include start and stop value. With shape of 0-D.
Outputs:
Tensor, With type same as `start`. The shape is 1-D with length of `num`.
Examples:
>>> linspace = nn.LinSpace()
>>> start = Tensor(1, mindspore.float32)
>>> stop = Tensor(10, mindspore.float32)
>>> num = Tensor(5, mindspore.int32)
>>> output = linspace(start, stop, num)
>>> linspace = nn.LinSpace(1, 10, 5)
>>> output = linspace()
[1, 3.25, 5.5, 7.75, 10]
"""

3
mindspore/ops/operations/nn_ops.py
View File

@ -2711,7 +2711,7 @@ class ROIAlign(PrimitiveWithInfer):
Examples:
>>> input_tensor = Tensor(np.array([[[[1., 2.], [3., 4.]]]]), mindspore.float32)
>>> rois = Tensor(np.array([[0, 0.2, 0.3, 0.2, 0.3]]), mindspore.float32)
>>> roi_align = P.ROIAlign(1, 1, 0.5, 2)
>>> roi_align = P.ROIAlign(2, 2, 0.5, 2)
>>> output_tensor = roi_align(input_tensor, rois)
>>> assert output_tensor == Tensor(np.array([[[[2.15]]]]), mindspore.float32)
"""
@ -4980,4 +4980,5 @@ class LRN(PrimitiveWithInfer):
return x_dtype
def infer_shape(self, x_shape):
validator.check_integer("x_shape", len(x_shape), 4, Rel.EQ, self.name)
return x_shape