array_length¶

paddle.fluid.layers. array_length ( array ) ¶

该OP用于获取输入数组 cn_api_fluid_LoDTensorArray 的长度。可以与 array_read 、 array_write 、 While OP结合使用，实现LoDTensorArray的遍历与读写。

参数¶

array (LoDTensorArray) - 输入的数组LoDTensorArray

返回¶

shape为[1]的1-D Tensor，表示数组LoDTensorArray的长度，数据类型为int64

返回类型¶

Variable

代码示例¶

          import paddle.fluid as fluid
tmp = fluid.layers.zeros(shape=[10], dtype='int32')
i = fluid.layers.fill_constant(shape=[1], dtype='int64', value=10)
#tmp是shape为[10]的Tensor，将tmp写入到数组arr下标为10的位置，arr的长度因此为11
arr = fluid.layers.array_write(tmp, i=i)
#查看arr的长度
arr_len = fluid.layers.array_length(arr)

#可以通过executor打印出LoDTensorArray的长度
input = fluid.layers.Print(arr_len, message="The length of LoDTensorArray:")
main_program = fluid.default_main_program()
exe = fluid.Executor(fluid.CPUPlace())
exe.run(main_program)

         

运行输出

COPY-FROM: paddle.fluid.layers.array_length