Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
e1b2d31d
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 搜索 >>
提交
e1b2d31d
编写于
4月 02, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
test(mge): skip some doctests having different results on gpu
GitOrigin-RevId: 66e5db5c891ca7373bd642a79974fef4c4a8ae03
上级
20b67b29
变更
4
显示空白变更内容
内联
并排
Showing
4 changed file
with
11 addition
and
8 deletion
+11
-8
python_module/megengine/functional/nn.py
python_module/megengine/functional/nn.py
+4
-3
python_module/megengine/functional/tensor.py
python_module/megengine/functional/tensor.py
+4
-2
python_module/megengine/module/conv.py
python_module/megengine/module/conv.py
+1
-1
python_module/megengine/random/distribution.py
python_module/megengine/random/distribution.py
+2
-2
未找到文件。
python_module/megengine/functional/nn.py
浏览文件 @
e1b2d31d
...
...
@@ -741,11 +741,11 @@ def dropout(inp: Tensor, drop_prob: float, rescale: bool = True) -> Tensor:
.. testcode::
import numpy as np
from megengine import tensor
import megengine as mge
import megengine.functional as F
from megengine
.random import manual_seed
from megengine
import tensor
manual_seed(0)
data = tensor(np.ones(10, dtype=np.float32))
out = F.dropout(data, 1./3.)
print(out.numpy())
...
...
@@ -753,6 +753,7 @@ def dropout(inp: Tensor, drop_prob: float, rescale: bool = True) -> Tensor:
Outputs:
.. testoutput::
:options: +SKIP
[1.5 1.5 0. 1.5 1.5 1.5 1.5 1.5 1.5 1.5]
...
...
python_module/megengine/functional/tensor.py
浏览文件 @
e1b2d31d
...
...
@@ -249,6 +249,7 @@ def scatter(inp: Tensor, axis: int, index: Tensor, source: Tensor) -> Tensor:
import numpy as np
import megengine.functional as F
from megengine.core import tensor
inp = tensor(np.zeros(shape=(3,5),dtype=np.float32))
source = tensor([[0.9935,0.9465,0.2256,0.8926,0.4396],[0.7723,0.0718,0.5939,0.357,0.4576]])
index = tensor([[0,2,0,2,1],[2,0,0,1,2]])
...
...
@@ -258,6 +259,7 @@ def scatter(inp: Tensor, axis: int, index: Tensor, source: Tensor) -> Tensor:
Outputs:
.. testoutput::
:options: +SKIP
[[0.9935 0.0718 0.5939 0. 0. ]
[0. 0. 0. 0.357 0.4396]
...
...
python_module/megengine/module/conv.py
浏览文件 @
e1b2d31d
...
...
@@ -115,7 +115,7 @@ class Conv2d(_ConvNd):
and there would be an extra dimension at the beginning of the weight's
shape. Specifically, the shape of weight would be ``(groups,
out_channel // groups, in_channels // groups, *kernel_size)``.
:param bias: wether to add a bias onto the result of convolution. Default:
:param bias: w
h
ether to add a bias onto the result of convolution. Default:
True
:param conv_mode: Supports `CROSS_CORRELATION` or `CONVOLUTION`. Default:
`CROSS_CORRELATION`.
...
...
python_module/megengine/random/distribution.py
浏览文件 @
e1b2d31d
...
...
@@ -42,11 +42,11 @@ def gaussian(
import megengine as mge
import megengine.random as rand
rand.manual_seed(0)
x = rand.gaussian((2, 2), mean=0, std=1)
print(x.numpy())
.. testoutput::
:options: +SKIP
[[-0.20235455 -0.6959438 ]
[-1.4939808 -1.5824696 ]]
...
...
@@ -79,11 +79,11 @@ def uniform(
import megengine as mge
import megengine.random as rand
rand.manual_seed(0)
x = rand.uniform((2, 2))
print(x.numpy())
.. testoutput::
:options: +SKIP
[[0.76901674 0.70496535]
[0.09365904 0.62957656]]
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录