create_random_int_lodtensor
create_random_int_lodtensor
(recursive_seq_lens, base_shape, place, low, high)[源代码]
具体实现方法如下:
- 基于序列长度
recursive_seq_lens
和 产生返回值的维度。返回值的第一维等于序列总长度,其余维度为base_shape
。
- 参数:
- recursive_seq_lens (list[list[int]]) - 基于序列长度的LoD信息。
- place (CPUPlace|CUDAPlace) - 表示返回的LoDTensor存储在CPU或GPU place中。
- low (int) - 随机整数的下限值。
- high (int) - 随机整数的上限值,必须大于或等于low。
返回: 包含随机整数数据信息和序列长度信息的LoDTensor,数值范围在[low, high]之间。
代码示例