Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
夏有凉风,冬有雪
Paddle
提交
7c0facd1
P
Paddle
项目概览
夏有凉风,冬有雪
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
7c0facd1
编写于
10月 15, 2017
作者:
Q
qijun
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
init
上级
17b4cea4
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
118 addition
and
0 deletion
+118
-0
paddle/framework/scope.h
paddle/framework/scope.h
+62
-0
paddle/pybind/pybind.cc
paddle/pybind/pybind.cc
+6
-0
python/paddle/v2/framework/tests/test_feed_fetch_method.py
python/paddle/v2/framework/tests/test_feed_fetch_method.py
+50
-0
未找到文件。
paddle/framework/scope.h
浏览文件 @
7c0facd1
...
@@ -18,8 +18,10 @@ limitations under the License. */
...
@@ -18,8 +18,10 @@ limitations under the License. */
#include <string>
#include <string>
#include <unordered_map>
#include <unordered_map>
#include "paddle/framework/lod_tensor.h"
#include "paddle/framework/variable.h"
#include "paddle/framework/variable.h"
#include "paddle/platform/macros.h"
#include "paddle/platform/macros.h"
#include "paddle/platform/place.h"
namespace
paddle
{
namespace
paddle
{
namespace
framework
{
namespace
framework
{
...
@@ -75,5 +77,65 @@ class Scope {
...
@@ -75,5 +77,65 @@ class Scope {
framework
::
Scope
&
GetGlobalScope
();
framework
::
Scope
&
GetGlobalScope
();
// template <typename T>
// void SetFeedVariable(const std::vector<T>& input, const Lod& lod,
// const std::vector<int64_t>& dims,
// const std::string& var_name, size_t index) {
// Variable* g_feed_value = GetGlobalScope().Var("var_name");
// // feed variable holds vector<LodTensor>
// auto& feed_inputs =
// *(g_feed_value->GetMutable<
// std::vector<paddle::framework::LoDTensor>>());
// if (index >= feed_inputs.size()) {
// feed_inputs.resize(index);
// }
// // copy tensor
// T* dst = feed_inputs[index].mutable_data<T>(make_ddim(dims),
// platform::CPUPlace());
// memcpy(dst, inputs[i].data(), inputs[i].size() * sizeof(T));
// // copy lod
// feed_inputs[index].set_lod(lod);
// }
template
<
typename
T
>
void
SetFeedVariable
(
const
LoDTensor
&
input
,
const
std
::
string
&
var_name
,
size_t
index
)
{
std
::
cout
<<
"into SetFeedVariable"
<<
std
::
endl
;
std
::
cout
<<
var_name
<<
std
::
endl
;
std
::
cout
<<
index
<<
std
::
endl
;
Variable
*
g_feed_value
=
GetGlobalScope
().
Var
(
var_name
);
auto
&
feed_inputs
=
*
(
g_feed_value
->
GetMutable
<
std
::
vector
<
paddle
::
framework
::
LoDTensor
>>
());
if
(
index
>=
feed_inputs
.
size
())
{
feed_inputs
.
resize
(
index
+
1
);
}
// shared data with input tensor
feed_inputs
[
index
].
ShareDataWith
<
T
>
(
input
);
// set lod
feed_inputs
[
index
].
set_lod
(
input
.
lod
());
}
// template <typename T>
// std::vector<T> GetFetchVariable(const std::string& var_name, size_t index) {
// Variable* g_fetch_value = GetGlobalScope().Var(var_name);
// auto& fetch_outputs =
// *(g_fetch_value->GetMutable<
// std::vector<paddle::framework::LoDTensor>>());
// std::vector<T> result;
// result.resize(fetch_outputs[index].numel());
// memcpy(result.data(), fetch_outputs[i].data<T>(),
// fetch_outputs[i].numel() * sizeof(T));
// }
template
<
typename
T
>
LoDTensor
&
GetFetchVariable
(
const
std
::
string
&
var_name
,
size_t
index
)
{
Variable
*
g_fetch_value
=
GetGlobalScope
().
Var
(
var_name
);
auto
&
fetch_outputs
=
*
(
g_fetch_value
->
GetMutable
<
std
::
vector
<
paddle
::
framework
::
LoDTensor
>>
());
std
::
cout
<<
"into GetFetchVariable"
<<
std
::
endl
;
PADDLE_ENFORCE_LT
(
index
,
fetch_outputs
.
size
());
return
fetch_outputs
[
index
];
}
}
// namespace framework
}
// namespace framework
}
// namespace paddle
}
// namespace paddle
paddle/pybind/pybind.cc
浏览文件 @
7c0facd1
...
@@ -394,6 +394,12 @@ All parameter, weight, gradient are variables in Paddle.
...
@@ -394,6 +394,12 @@ All parameter, weight, gradient are variables in Paddle.
m
.
def
(
"unique_integer"
,
UniqueIntegerGenerator
);
m
.
def
(
"unique_integer"
,
UniqueIntegerGenerator
);
m
.
def
(
"is_compile_gpu"
,
IsCompileGPU
);
m
.
def
(
"is_compile_gpu"
,
IsCompileGPU
);
m
.
def
(
"set_feed_variable"
,
SetFeedVariable
<
float
>
);
// m.def("set_feed_variable", SetFeedVariable<double>);
// m.def("set_feed_variable", SetFeedVariable<int>);
m
.
def
(
"get_fetch_variable"
,
GetFetchVariable
<
float
>
);
// m.def("get_fetch_variable", GetFetchVariable<double>);
// m.def("get_fetch_variable", GetFetchVariable<int>);
BindProgramDesc
(
m
);
BindProgramDesc
(
m
);
BindBlockDesc
(
m
);
BindBlockDesc
(
m
);
...
...
python/paddle/v2/framework/tests/test_feed_fetch_method.py
0 → 100644
浏览文件 @
7c0facd1
import
paddle.v2.framework.core
as
core
import
unittest
import
numpy
as
np
# class TestFeedFetch(unittest.TestCase):
# def test_feed_fetch(self):
# place = core.CPUPlace()
# input_tensor = core.LoDTensor([[0, 2, 4]])
# input_tensor.set_dims([4, 4, 6])
# input_tensor.alloc_int(place)
# input_array = np.array(input_tensor)
# input_array[0, 0, 0] = 3
# input_array[3, 3, 5] = 10
# input_tensor.set(input_array, place)
# core.set_feed_variable(input_tensor, "feed", 0)
# output_tensor = core.get_fetch_variable("feed", 0)
# print type(output_tensor)
# output_lod = output_tensor.lod()
# print type(output_lod)
# print output_lod[0]
# print output_lod[0][0]
# print output_lod[0][1]
# print output_lod[0][2]
# # self.assertEqual(0, output_lod[0][0])
# # self.assertEqual(0, output_lod[0][0])
# # self.assertEqual(2, output_lod[0][1])
# # self.assertEqual(4, output_lod[0][2])
# # output_array = np.array(output_tensor)
# # self.assertEqual(3, output_array[0, 0, 0])
# # self.assertEqual(10, output_array[3, 3, 5]);
class
TestFeedFetch
(
unittest
.
TestCase
):
def
test_feed_fetch
(
self
):
place
=
core
.
CPUPlace
()
input_tensor
=
core
.
LoDTensor
([[
0
,
2
,
4
]])
input_tensor
.
set_dims
([
4
,
4
,
6
])
input_tensor
.
alloc_float
(
place
)
input_array
=
np
.
array
(
input_tensor
)
input_array
[
0
,
0
,
0
]
=
3
input_array
[
3
,
3
,
5
]
=
10
input_tensor
.
set
(
input_array
,
place
)
if
__name__
==
"__main__"
:
unittest
.
main
()
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录