create_lod_tensor
create_lod_tensor
(data, recursive_seq_lens, place)[源代码]
具体实现方法如下:
- 检查基于序列长度的LoD(length-based LoD),即参数中的 是否正确。
- 根据place参数,把所提供的 (numpy数组、list或LoDTensor)的数据复制到CPU或GPU上。
- 将基于偏移量的LoD设置到输出的LoDTensor中。
那么, data
为一个维度为(5, 1)的numpy整数数组; 为[[2, 3]],表示每个句子含的单词个数。在该接口内部,基于序列长度的 recursive_seq_lens
[[2, 3]]会转换为为基于偏移量的LoD [[0, 2, 5]]。
- 参数:
返回: 包含数据信息和序列长度信息的LoDTensor。
代码示例