Skip to content
This repository was archived by the owner on Jan 24, 2024. It is now read-only.

Commit 6defa44

Browse files
Create data_0.py
add the first fluid.layers.data api
1 parent f471f82 commit 6defa44

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed
Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
# Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
# api: paddle.fluid.data
16+
# env: local
17+
# device: gpu
18+
# text:feed-gpu-data
19+
20+
import paddle.fluid as fluid
21+
import numpy as np
22+
23+
def gen_data():
24+
return {"x": np.ones((1, 32)).astype('float32')}
25+
26+
input_x = fluid.layers.data(name="x", shape=[32], dtype='float32')
27+
place = fluid.CUDAPlace(0)
28+
exe = fluid.Executor(place)
29+
input_val = exe.run(feed=gen_data(),
30+
fetch_list=[input_x.name])
31+
print(input_val)
32+
# [array([[1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
33+
# 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.]],
34+
# dtype=float32)]

0 commit comments

Comments
 (0)