关于序列数据输入后打印tensor dim的问题
Created by: 333caowei
我有一个id list输入给模型,每个id代表一个词,例如[9,8,3,4,5]
query_data = fluid.layers.data(
name="query",
shape=[1],
dtype='int64',
lod_level=1)
query_emb = fluid.layers.embedding(
input=query_data,
size=[VOCAB_DICT_SIZE, 128],
is_sparse=IS_SPARSE)
print query_emb
可是print query_emb信息如下:
type {
type: LOD_TENSOR
lod_tensor {
tensor {
data_type: FP32
dims: -1
dims: 128
}
lod_level: 1
}
}
persistable: false
为什么query_emb的dim是2维的呢,第一维-1表示batchsize,第二维128表示向量dim,应该还有第三维表示序列长度才对