create_random_int_lodtensor_cn.rst 1.8 KB
Newer Older
H
Hao Wang 已提交
1 2 3 4 5 6
.. _cn_api_fluid_create_random_int_lodtensor:


create_random_int_lodtensor
-------------------------------

7 8
**注意:该API仅支持【静态图】模式**

H
Hao Wang 已提交
9 10
.. py:function:: paddle.fluid.create_random_int_lodtensor(recursive_seq_lens, base_shape, place, low, high)

11
创建一个包含随机整数的LoDTensor。
H
Hao Wang 已提交
12

13
具体实现方法如下:
H
Hao Wang 已提交
14

15
1. 基于序列长度 :code:`recursive_seq_lens` 和 :code:`base_shape` 产生返回值的维度。返回值的第一维等于序列总长度,其余维度为 :code:`base_shape` 。
H
Hao Wang 已提交
16

17
2. 创建一个包含随机整数的numpy数组,并作为 :code:`data` 参数传入 :ref:`cn_api_fluid_create_lod_tensor` 接口中创建LoDTensor返回。
H
Hao Wang 已提交
18

19 20
假设我们想创建一个LoDTensor表示序列信息,共包含2个序列,维度分别为[2, 30]和[3, 30],那么序列长度 :code:`recursive_seq_lens` 传入[[2, 3]],:code:`base_shape` 传入[30](即除了序列长度以外的维度)。
最后返回的LoDTensor的维度为[5, 30],其中第一维5为序列总长度,其余维度为 :code:`base_shape` 。
H
Hao Wang 已提交
21 22

参数:
23
    - **recursive_seq_lens** (list[list[int]]) - 基于序列长度的LoD信息。
24
    - **base_shape** (list[int]) - 除第一维以外输出结果的维度信息。
25 26 27 28 29
    - **place** (CPUPlace|CUDAPlace) - 表示返回的LoDTensor存储在CPU或GPU place中。
    - **low** (int) - 随机整数的下限值。
    - **high** (int) - 随机整数的上限值,必须大于或等于low。

返回: 包含随机整数数据信息和序列长度信息的LoDTensor,数值范围在[low, high]之间。
H
Hao Wang 已提交
30

31
返回类型: LoDTensor
H
Hao Wang 已提交
32 33 34 35 36 37 38 39

**代码示例**

.. code-block:: python

        import paddle.fluid as fluid
     
        t = fluid.create_random_int_lodtensor(recursive_seq_lens=[[2, 3]],base_shape=[30], place=fluid.CPUPlace(), low=0, high=10)
40
        print(t.shape()) # [5, 30]
H
Hao Wang 已提交
41