Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
0d22e64f
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
0d22e64f
编写于
7月 07, 2020
作者:
J
jiangjinsheng
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix LinSpace doc
上级
7304f024
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
8 addition
and
11 deletion
+8
-11
mindspore/nn/layer/math.py
mindspore/nn/layer/math.py
+6
-10
mindspore/ops/operations/nn_ops.py
mindspore/ops/operations/nn_ops.py
+2
-1
未找到文件。
mindspore/nn/layer/math.py
浏览文件 @
0d22e64f
...
@@ -132,23 +132,19 @@ class Range(Cell):
...
@@ -132,23 +132,19 @@ class Range(Cell):
class
LinSpace
(
Cell
):
class
LinSpace
(
Cell
):
r
"""
r
"""
Generates values in an interval.
And return the corresponding interpolation accroding to assist.
Generates values in an interval.
Args:
Args:
- **start** (Union[int, float]) - The start of interval, 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.
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.
num (int): ticks number in the interval, the ticks include start and stop value. With shape of 0-D.
With shape of 0-D.
Outputs:
Outputs:
Tensor, With type same as `start`. The shape is 1-D with length of `num`.
Tensor, With type same as `start`. The shape is 1-D with length of `num`.
Examples:
Examples:
>>> linspace = nn.LinSpace()
>>> linspace = nn.LinSpace(1, 10, 5)
>>> start = Tensor(1, mindspore.float32)
>>> output = linspace()
>>> stop = Tensor(10, mindspore.float32)
>>> num = Tensor(5, mindspore.int32)
>>> output = linspace(start, stop, num)
[1, 3.25, 5.5, 7.75, 10]
[1, 3.25, 5.5, 7.75, 10]
"""
"""
...
...
mindspore/ops/operations/nn_ops.py
浏览文件 @
0d22e64f
...
@@ -2711,7 +2711,7 @@ class ROIAlign(PrimitiveWithInfer):
...
@@ -2711,7 +2711,7 @@ class ROIAlign(PrimitiveWithInfer):
Examples:
Examples:
>>> input_tensor = Tensor(np.array([[[[1., 2.], [3., 4.]]]]), mindspore.float32)
>>> 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)
>>> 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)
>>> output_tensor = roi_align(input_tensor, rois)
>>> assert output_tensor == Tensor(np.array([[[[2.15]]]]), mindspore.float32)
>>> assert output_tensor == Tensor(np.array([[[[2.15]]]]), mindspore.float32)
"""
"""
...
@@ -4980,4 +4980,5 @@ class LRN(PrimitiveWithInfer):
...
@@ -4980,4 +4980,5 @@ class LRN(PrimitiveWithInfer):
return
x_dtype
return
x_dtype
def
infer_shape
(
self
,
x_shape
):
def
infer_shape
(
self
,
x_shape
):
validator
.
check_integer
(
"x_shape"
,
len
(
x_shape
),
4
,
Rel
.
EQ
,
self
.
name
)
return
x_shape
return
x_shape
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录