Tensor_cn.rst 1.4 KB
Newer Older
H
Hao Wang 已提交
1 2 3 4 5 6 7
.. _cn_api_fluid_Tensor:

Tensor
-------------------------------

.. py:function:: paddle.fluid.Tensor

8
Tensor用于表示多维张量,可以通过 ``np.array(tensor)`` 方法转换为numpy.ndarray。
H
Hao Wang 已提交
9

10
**示例代码**
H
Hao Wang 已提交
11

12
.. code-block:: python
H
Hao Wang 已提交
13

14
      import paddle.fluid as fluid
H
Hao Wang 已提交
15

16
      t = fluid.Tensor()
H
Hao Wang 已提交
17

18
.. py:method::  set(array, place, zero_copy=False)
H
Hao Wang 已提交
19

20
该接口根据输入的numpy array和设备place,设置Tensor的数据。
H
Hao Wang 已提交
21

22 23 24
参数:
    - **array** (numpy.ndarray) - 要设置的numpy array,支持的数据类型为bool, float32, float64, int8, int32, int64, uint8, uint16。
    - **place** (CPUPlace|CUDAPlace|CUDAPinnedPlace) - 要设置的Tensor所在的设备。
25
    - **zero_copy** (bool,可选) - 是否与输入的numpy数组共享内存。此参数仅适用于CPUPlace。默认值为False。
26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56

返回:无。

**示例代码**

.. code-block:: python

            import paddle.fluid as fluid
            import numpy as np

            t = fluid.Tensor()
            t.set(np.ndarray([5, 30]), fluid.CPUPlace())

.. py:method::  shape(self: paddle.fluid.core_avx.Tensor) → List[int]

该接口返回Tensor的shape。

返回:Tensor的shape。

返回类型:List[int] 。

**示例代码**

.. code-block:: python

            import paddle.fluid as fluid
            import numpy as np

            t = fluid.Tensor()
            t.set(np.ndarray([5, 30]), fluid.CPUPlace())
            print(t.shape())  # [5, 30]