Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
c8920945
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
403
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看板
提交
c8920945
编写于
11月 05, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
ci(mge): remove blank line for windows doctest
GitOrigin-RevId: 09cf6518e8cf96113f455a6222ba17f271143bb4
上级
e082e277
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
19 addition
and
55 deletion
+19
-55
imperative/python/megengine/functional/tensor.py
imperative/python/megengine/functional/tensor.py
+12
-29
imperative/python/megengine/module/embedding.py
imperative/python/megengine/module/embedding.py
+7
-26
未找到文件。
imperative/python/megengine/functional/tensor.py
浏览文件 @
c8920945
...
...
@@ -208,25 +208,16 @@ def broadcast_to(inp: Tensor, shape: Union[int, Iterable[int]]) -> Tensor:
from megengine import tensor
import megengine.functional as F
data = tensor(np.arange(0,
6, dtype=np.float32).reshape(2,
3))
out = F.broadcast_to(data, (
4,
2, 3))
data = tensor(np.arange(0,
3, dtype=np.float32).reshape(
3))
out = F.broadcast_to(data, (2, 3))
print(out.numpy())
Outputs:
.. testoutput::
[[[0. 1. 2.]
[3. 4. 5.]]
[[0. 1. 2.]
[3. 4. 5.]]
[[0. 1. 2.]
[3. 4. 5.]]
[[0. 1. 2.]
[3. 4. 5.]]]
[[0. 1. 2.]
[0. 1. 2.]]
"""
return
_broadcast
(
inp
,
shape
)
...
...
@@ -298,8 +289,8 @@ def stack(inps, axis=0, device=None):
from megengine import tensor
import megengine.functional as F
x1 = tensor(np.arange(0,
6, dtype=np.float32).reshape((2,
3)))
x2 = tensor(np.arange(6,
12, dtype=np.float32).reshape((2,
3)))
x1 = tensor(np.arange(0,
3, dtype=np.float32).reshape((
3)))
x2 = tensor(np.arange(6,
9, dtype=np.float32).reshape((
3)))
out = F.stack([x1, x2], axis=0)
print(out.numpy())
...
...
@@ -307,11 +298,8 @@ def stack(inps, axis=0, device=None):
.. testoutput::
[[[ 0. 1. 2.]
[ 3. 4. 5.]]
[[ 6. 7. 8.]
[ 9. 10. 11.]]]
[[0. 1. 2.]
[6. 7. 8.]]
"""
if
len
(
inps
)
>
0
and
not
isinstance
(
inps
[
0
].
shape
,
inps
[
0
].
__class__
):
...
...
@@ -751,21 +739,16 @@ def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor:
from megengine import tensor
import megengine.functional as F
x = tensor(np.arange(12, dtype=np.int32))
out = F.reshape(x, (3,
2, 2
))
out = F.reshape(x, (3,
4
))
print(out.numpy())
Outputs:
.. testoutput::
[[[ 0 1]
[ 2 3]]
[[ 4 5]
[ 6 7]]
[[ 8 9]
[10 11]]]
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
"""
return
inp
.
reshape
(
target_shape
)
...
...
imperative/python/megengine/module/embedding.py
浏览文件 @
c8920945
...
...
@@ -38,10 +38,10 @@ class Embedding(Module):
import numpy as np
import megengine as mge
import megengine.module as M
weight = mge.tensor(np.array([(1.2,2.3,3.4,4.5,5.6)
,(0.1,1.1,2.1,3.1,4.1)
], dtype=np.float32))
data = mge.tensor(np.array([(0,
1,1),(1,0,1),(0,0,1
)], dtype=np.int32))
weight = mge.tensor(np.array([(1.2,2.3,3.4,4.5,5.6)], dtype=np.float32))
data = mge.tensor(np.array([(0,
0
)], dtype=np.int32))
embedding = M.Embedding(
2
, 5, initial_weight=weight)
embedding = M.Embedding(
1
, 5, initial_weight=weight)
output = embedding(data)
with np.printoptions(precision=6):
print(output.numpy())
...
...
@@ -51,16 +51,7 @@ class Embedding(Module):
.. testoutput::
[[[1.2 2.3 3.4 4.5 5.6]
[0.1 1.1 2.1 3.1 4.1]
[0.1 1.1 2.1 3.1 4.1]]
[[0.1 1.1 2.1 3.1 4.1]
[1.2 2.3 3.4 4.5 5.6]
[0.1 1.1 2.1 3.1 4.1]]
[[1.2 2.3 3.4 4.5 5.6]
[1.2 2.3 3.4 4.5 5.6]
[0.1 1.1 2.1 3.1 4.1]]]
[1.2 2.3 3.4 4.5 5.6]]]
"""
...
...
@@ -134,8 +125,8 @@ class Embedding(Module):
import numpy as np
import megengine as mge
import megengine.module as M
weight = mge.tensor(np.array([(1.2,2.3,3.4,4.5,5.6)
,(0.1,1.1,2.1,3.1,4.1)
], dtype=np.float32))
data = mge.tensor(np.array([(0,
1,1),(1,0,1),(0,0,1
)], dtype=np.int32))
weight = mge.tensor(np.array([(1.2,2.3,3.4,4.5,5.6)], dtype=np.float32))
data = mge.tensor(np.array([(0,
0
)], dtype=np.int32))
embedding = M.Embedding.from_pretrained(weight, freeze=False)
output = embedding(data)
...
...
@@ -146,17 +137,7 @@ class Embedding(Module):
.. testoutput::
[[[1.2 2.3 3.4 4.5 5.6]
[0.1 1.1 2.1 3.1 4.1]
[0.1 1.1 2.1 3.1 4.1]]
[[0.1 1.1 2.1 3.1 4.1]
[1.2 2.3 3.4 4.5 5.6]
[0.1 1.1 2.1 3.1 4.1]]
[[1.2 2.3 3.4 4.5 5.6]
[1.2 2.3 3.4 4.5 5.6]
[0.1 1.1 2.1 3.1 4.1]]]
[1.2 2.3 3.4 4.5 5.6]]]
"""
embeddings_shape
=
embeddings
.
shape
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录