Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
13272eaa
MegEngine
项目概览
MegEngine 天元
/
MegEngine
9 个月 前同步成功
通知
392
Star
4702
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看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
13272eaa
编写于
11月 18, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(mge/trace): fix random op in symbolic trace
GitOrigin-RevId: 9a851cd177119ee43155b831e9622ce342423090
上级
1fed5929
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
25 addition
and
2 deletion
+25
-2
imperative/python/megengine/random/distribution.py
imperative/python/megengine/random/distribution.py
+4
-2
imperative/python/test/unit/test_tracing.py
imperative/python/test/unit/test_tracing.py
+21
-0
未找到文件。
imperative/python/megengine/random/distribution.py
浏览文件 @
13272eaa
...
...
@@ -52,7 +52,8 @@ def normal(
size
=
(
1
,)
seed
=
_random_seed_generator
().
__next__
()
op
=
GaussianRNG
(
seed
=
seed
,
mean
=
mean
,
std
=
std
)
size
=
Tensor
(
size
,
dtype
=
"int32"
)
_ref
=
Tensor
([],
dtype
=
"int32"
)
size
=
utils
.
astensor1d
(
size
,
_ref
,
dtype
=
"int32"
)
(
output
,)
=
apply
(
op
,
size
)
return
output
...
...
@@ -93,7 +94,8 @@ def uniform(
size
=
(
1
,)
seed
=
_random_seed_generator
().
__next__
()
op
=
UniformRNG
(
seed
=
seed
)
size
=
Tensor
(
size
,
dtype
=
"int32"
)
_ref
=
Tensor
([],
dtype
=
"int32"
)
size
=
utils
.
astensor1d
(
size
,
_ref
,
dtype
=
"int32"
)
(
output
,)
=
apply
(
op
,
size
)
return
low
+
(
high
-
low
)
*
output
imperative/python/test/unit/test_tracing.py
浏览文件 @
13272eaa
...
...
@@ -23,6 +23,7 @@ from megengine.core.tensor.core import apply
from
megengine.core.tensor.raw_tensor
import
as_raw_tensor
from
megengine.functional
import
exp
,
log
from
megengine.jit
import
exclude_from_trace
,
trace
from
megengine.random
import
normal
,
uniform
def
test_trace
():
...
...
@@ -431,3 +432,23 @@ def test_slice():
y
=
f
(
x
)
np
.
testing
.
assert_array_equal
(
y
.
numpy
(),
x
.
numpy
()[:,
1
::
2
])
y
+
y
def
test_random
():
def
run_test
(
op
):
for
symbolic_shape
in
[
True
,
False
]:
@
trace
(
symbolic
=
True
,
symbolic_shape
=
symbolic_shape
)
def
f
():
out
=
op
(
size
=
[
10
,
10
])
out_shape
=
out
.
shape
assert
out_shape
is
not
None
if
not
isinstance
(
out_shape
,
tuple
):
assert
out
.
shape
.
numpy
()
is
not
None
return
out
for
_
in
range
(
3
):
f
()
run_test
(
uniform
)
run_test
(
normal
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录