Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
e42679b5
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
404
Star
4705
Fork
582
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
e42679b5
编写于
10月 09, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
feat(mge): do not export F.nn.interpolate
GitOrigin-RevId: 031c6555c0a4c190594d96e47d989da5cd5df62f
上级
fd802e97
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
10 addition
and
11 deletion
+10
-11
imperative/python/megengine/functional/nn.py
imperative/python/megengine/functional/nn.py
+2
-3
imperative/python/test/unit/functional/test_functional.py
imperative/python/test/unit/functional/test_functional.py
+8
-8
未找到文件。
imperative/python/megengine/functional/nn.py
浏览文件 @
e42679b5
...
...
@@ -36,7 +36,6 @@ __all__ = [
"dot"
,
"dropout"
,
"indexing_one_hot"
,
"interpolate"
,
"leaky_relu"
,
"linear"
,
"local_conv2d"
,
...
...
@@ -1112,9 +1111,9 @@ def interpolate(
import megengine.functional as F
x = tensor(np.arange(1, 5, dtype=np.float32).reshape(1, 1, 2, 2))
out = F.interpolate(x, [4, 4], align_corners=False)
out = F.
nn.
interpolate(x, [4, 4], align_corners=False)
print(out.numpy())
out2 = F.interpolate(x, scale_factor=2.)
out2 = F.
nn.
interpolate(x, scale_factor=2.)
np.testing.assert_allclose(out.numpy(), out2.numpy())
Outputs:
...
...
imperative/python/test/unit/functional/test_functional.py
浏览文件 @
e42679b5
...
...
@@ -101,8 +101,8 @@ def test_interpolate():
def
linear_interpolate
():
inp
=
tensor
(
np
.
arange
(
1
,
3
,
dtype
=
np
.
float32
).
reshape
(
1
,
1
,
2
))
out
=
F
.
interpolate
(
inp
,
scale_factor
=
2.0
,
mode
=
"LINEAR"
)
out2
=
F
.
interpolate
(
inp
,
4
,
mode
=
"LINEAR"
)
out
=
F
.
nn
.
interpolate
(
inp
,
scale_factor
=
2.0
,
mode
=
"LINEAR"
)
out2
=
F
.
nn
.
interpolate
(
inp
,
4
,
mode
=
"LINEAR"
)
np
.
testing
.
assert_allclose
(
out
.
numpy
(),
np
.
array
([[[
1.0
,
1.25
,
1.75
,
2.0
]]],
dtype
=
np
.
float32
)
...
...
@@ -114,16 +114,16 @@ def test_interpolate():
def
many_batch_interpolate
():
inp
=
tensor
(
np
.
arange
(
1
,
9
,
dtype
=
np
.
float32
).
reshape
(
2
,
1
,
2
,
2
))
out
=
F
.
interpolate
(
inp
,
[
4
,
4
])
out2
=
F
.
interpolate
(
inp
,
scale_factor
=
2.0
)
out
=
F
.
nn
.
interpolate
(
inp
,
[
4
,
4
])
out2
=
F
.
nn
.
interpolate
(
inp
,
scale_factor
=
2.0
)
np
.
testing
.
assert_allclose
(
out
.
numpy
(),
out2
.
numpy
())
def
assign_corner_interpolate
():
inp
=
tensor
(
np
.
arange
(
1
,
5
,
dtype
=
np
.
float32
).
reshape
(
1
,
1
,
2
,
2
))
out
=
F
.
interpolate
(
inp
,
[
4
,
4
],
align_corners
=
True
)
out2
=
F
.
interpolate
(
inp
,
scale_factor
=
2.0
,
align_corners
=
True
)
out
=
F
.
nn
.
interpolate
(
inp
,
[
4
,
4
],
align_corners
=
True
)
out2
=
F
.
nn
.
interpolate
(
inp
,
scale_factor
=
2.0
,
align_corners
=
True
)
np
.
testing
.
assert_allclose
(
out
.
numpy
(),
out2
.
numpy
())
...
...
@@ -131,13 +131,13 @@ def test_interpolate():
inp
=
tensor
(
np
.
arange
(
1
,
5
,
dtype
=
np
.
float32
).
reshape
(
1
,
1
,
2
,
2
))
with
pytest
.
raises
(
ValueError
):
F
.
interpolate
(
inp
,
scale_factor
=
2.0
,
mode
=
"LINEAR"
)
F
.
nn
.
interpolate
(
inp
,
scale_factor
=
2.0
,
mode
=
"LINEAR"
)
def
inappropriate_scale_linear_interpolate
():
inp
=
tensor
(
np
.
arange
(
1
,
3
,
dtype
=
np
.
float32
).
reshape
(
1
,
1
,
2
))
with
pytest
.
raises
(
ValueError
):
F
.
interpolate
(
inp
,
scale_factor
=
[
2.0
,
3.0
],
mode
=
"LINEAR"
)
F
.
nn
.
interpolate
(
inp
,
scale_factor
=
[
2.0
,
3.0
],
mode
=
"LINEAR"
)
linear_interpolate
()
many_batch_interpolate
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录