Tensor
paddle.fluid.
Tensor
()
Tensor用于表示多维张量,可以通过 np.array(tensor)
方法转换为numpy.ndarray。
示例代码
- import paddle.fluid as fluid
- 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。
返回:无。
示例代码
- import paddle.fluid as fluid
- import numpy as np
- t = fluid.Tensor()
- t.set(np.ndarray([5, 30]), fluid.CPUPlace())
shape(self: paddle.fluid.core_avx.Tensor) → List[int]
该接口返回Tensor的shape。
返回:Tensor的shape。
返回类型:List[int] 。
示例代码
- 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]