Tensor

paddle.fluid.Tensor ( )

Tensor用于表示多维张量，可以通过 np.array(tensor) 方法转换为numpy.ndarray。

示例代码

  1. import paddle.fluid as fluid
  3. t = fluid.Tensor()

set ( array, place, zero_copy=False )

该接口根据输入的numpy array和设备place，设置Tensor的数据。

参数：

  • array (numpy.ndarray) - 要设置的numpy array，支持的数据类型为bool, float32, float64, int8, int32, int64, uint8, uint16。

  • place (CPUPlace|CUDAPlace|CUDAPinnedPlace) - 要设置的Tensor所在的设备。

  • zero_copy (bool，可选) - 是否与输入的numpy数组共享内存。此参数仅适用于CPUPlace。默认值为False。

返回：无。

示例代码

  1. import paddle.fluid as fluid
  2. import numpy as np
  4. t = fluid.Tensor()
  5. t.set(np.ndarray([5, 30]), fluid.CPUPlace())

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

该接口返回Tensor的shape。

返回：Tensor的shape。

返回类型：List[int] 。

示例代码

  1. import paddle.fluid as fluid
  2. import numpy as np
  4. t = fluid.Tensor()
  5. t.set(np.ndarray([5, 30]), fluid.CPUPlace())
  6. print(t.shape())  # [5, 30]