Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Oneflow-Inc
oneflow
提交
9b30b7c9
O
oneflow
项目概览
Oneflow-Inc
/
oneflow
上一次同步 2 年多
通知
13
Star
2733
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
O
oneflow
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
未验证
提交
9b30b7c9
编写于
6月 02, 2021
作者:
X
Xiaoyu Zhang
提交者:
GitHub
6月 02, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add tensor detach python api (#5068)
* add argmax test * add tensor detach python api * delete unuse code
上级
58183f87
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
11 addition
and
0 deletion
+11
-0
oneflow/python/test/tensor/test_tensor.py
oneflow/python/test/tensor/test_tensor.py
+11
-0
未找到文件。
oneflow/python/test/tensor/test_tensor.py
浏览文件 @
9b30b7c9
...
...
@@ -619,6 +619,17 @@ class TestTensor(flow.unittest.TestCase):
np_out
=
np
.
array
([[
1.0000
,
0.3139
],
[
0.3898
,
1.0000
],
[
0.0478
,
1.0000
]])
test_case
.
assertTrue
(
np
.
allclose
(
of_out
.
numpy
(),
np_out
))
@
unittest
.
skipIf
(
not
flow
.
unittest
.
env
.
eager_execution_enabled
(),
"numpy doesn't work in lazy mode"
,
)
def
test_tensor_detach
(
test_case
):
shape
=
(
2
,
3
,
4
,
5
)
x
=
flow
.
Tensor
(
np
.
random
.
randn
(
*
shape
),
dtype
=
flow
.
float32
,
requires_grad
=
True
,
)
test_case
.
assertTrue
(
np
.
allclose
(
x
.
detach
().
numpy
(),
x
.
numpy
()))
if
__name__
==
"__main__"
:
unittest
.
main
()
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录