见代码
import paddle.fluid as fluid import numpy as np # data, embedding data = fluid.data(name="sequence", shape=[-1, 1], dtype="int64") emb_1 = fluid.embedding(input=data, size=[12, 8]) # data 赋值 feed_data = x = np.array([[1], [2]]) exe = fluid.Executor(fluid.CPUPlace()) exe.run(fluid.default_startup_program()) out = exe.run(feed = {'sequence': feed_data}, fetch_list = [emb_1]) print(out)官方文档 https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/layers_cn/data_cn.html#data 中不推荐使用 paddle.fluid.layers.data ,因其在之后的版本中会被删除。请使用 paddle.fluid.data 。
官方文档 https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/layers_cn/embedding_cn.html#embedding 不推荐使用fluid.layers.embedding,此OP将在未来的版本中被移除!该OP要求输入Tensor shape的最后一维必须为1。推荐使用fluid. embedding 。
结果如下:
[array([[[ 0.16632503, 0.15989 , -0.458995 , 0.32499647, 0.03464723, -0.28962037, -0.19422564, 0.4503901 ]], [[-0.51977485, 0.02450722, -0.09264651, -0.5332941 , 0.3236062 , -0.25292158, -0.04155749, -0.259421 ]]], dtype=float32)]
