LoDTensorArray

class paddle.fluid.LoDTensorArray

LoDTensorArray是由LoDTensor组成的数组,支持”[]“运算符、len()函数和for迭代等。

代码示例

  1. import paddle.fluid as fluid
  2. arr = fluid.LoDTensorArray()

方法

append(self: paddle.fluid.core_avx.LoDTensorArray, tensor: paddle.fluid.core.LoDTensor) → None

该接口将LoDTensor追加到LoDTensorArray后。

参数

  • tensor (LoDTensor) - 追加的LoDTensor。

返回 无。

代码示例

  1. import paddle.fluid as fluid
  2. import numpy as np
  3. arr = fluid.LoDTensorArray()
  4. t = fluid.LoDTensor()
  5. t.set(np.ndarray([5, 30]), fluid.CPUPlace())
  6. arr.append(t)