dynamic_lstm输入问题
收藏
现在我获取到了一个变量,格式如下,其中-1代表batch_size,512表示句子长度,768表示embedding长度
name: "lower_bert_lower_bert_layer_norm_12.tmp_2"
type {
type: LOD_TENSOR
lod_tensor {
tensor {
data_type: FP32
dims: -1
dims: 512
dims: 768
}
lod_level: 0
}
}
persistable: false
想根据这个输入到dynamic_lstm中,但是不知道怎么构造一个lodtensor输入
0
收藏
请登录后评论
示例写的挺详细的吧,用全连接层先做线性映射