Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
afeixing77
Paddle
提交
994b52fc
P
Paddle
项目概览
afeixing77
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
994b52fc
编写于
3月 02, 2018
作者:
F
fengjiayi
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add layers for save/load op
上级
92974d41
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
69 addition
and
1 deletion
+69
-1
python/paddle/fluid/layers/nn.py
python/paddle/fluid/layers/nn.py
+69
-1
未找到文件。
python/paddle/fluid/layers/nn.py
浏览文件 @
994b52fc
...
...
@@ -3206,7 +3206,7 @@ def one_hot(input, depth):
operator.
Args:
input(
Tensor/LodTensor
): A Tensor/LodTensor of indices, last dimension must be 1.
input(
variable
): A Tensor/LodTensor of indices, last dimension must be 1.
depth(scalar): an interger defining the depth of the one hot dimension.
Returns:
...
...
@@ -3265,3 +3265,71 @@ def autoincreased_step_counter(counter_name=None, begin=1, step=1):
counter
.
stop_gradient
=
True
return
counter
def
save
(
x
,
file_path
,
overwrite
=
True
):
"""
Saves a variable as a file.
Args:
x(variable): The Tensor/LoDTensor to be saved.
file_path(str): The file path where the variable will be saved.
overwrite(bool): Whether or not cover the given file when it has already
existed. If it's set 'False' and the file is existed, a runtime
error will be thrown.
"""
helper
=
LayerHelper
(
"save"
,
**
locals
())
helper
.
append_op
(
type
=
"save"
,
inputs
=
{
"input"
:
x
},
outputs
=
{},
args
=
{
"file_path"
:
file_path
,
"overwrite"
:
overwrite
})
def
save_combine
(
x
,
file_path
,
overwrite
=
True
):
"""
Saves a variable as a file.
Args:
x(list): A list of Tensor/LoDTensor to be saved together in a single file.
file_path(str): The file path where variables will be saved.
overwrite(bool): Whether or not cover the given file when it has already
existed. If it's set 'False' and the file is existed, a runtime
error will be thrown.
"""
helper
=
LayerHelper
(
"save_combine"
,
**
locals
())
helper
.
append_op
(
type
=
"save_combine"
,
inputs
=
{
"input"
:
x
},
outputs
=
{},
args
=
{
"file_path"
:
file_path
,
"overwrite"
:
overwrite
})
def
load
(
out
,
file_path
):
"""
Args:
out(variable): The variable to be read from the disk file.
file_path(str): The path of the disk file.
"""
helper
=
LayerHelper
(
"load"
,
**
locals
())
helper
.
append_op
(
type
=
"load"
,
inputs
=
{},
output
=
{
"Out"
:
out
},
args
=
{
"file_path"
:
file_path
})
def
load_combine
(
out
,
file_path
):
"""
Args:
out(list): The list of variables to be read from the disk file.
file_path(str): The path of the disk file.
"""
helper
=
LayerHelper
(
"load_combine"
,
**
locals
())
helper
.
append_op
(
type
=
"load_combine"
,
inputs
=
{},
output
=
{
"Out"
:
out
},
args
=
{
"file_path"
:
file_path
})
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录